#include <opencv2\opencv.hpp>

#include <opencv2\dnn.hpp>

#include <stdlib.h>

#include <iostream>

#include <string.h>


using namespace cv;

using namespace std;

using namespace dnn;


int main()

{

	//读取图片

	Mat img = imread("F:\\VS15Workspace\\TensorflowTest\\data\\5.jpg", 0);

	

	//pb文件路径

	string pbfile = "F:\\VS15Workspace\\TensorflowTest\\mnist_model\\mnist.pb";


	//(3)读取Tensorflow模型文件

	Net net = readNetFromTensorflow(pbfile);


	//(4)创建输入数据

	Mat inputBlob = blobFromImage(img, 1.0, Size(28, 28), Scalar(), false, false);


	//(5)输入网络

	net.setInput(inputBlob);


	//(6)预测结果

	Mat pred = net.forward();


	//(7)输出结果

	Point maxLoc;

	minMaxLoc(pred, NULL, NULL, NULL, &maxLoc);

	cout << maxLoc.x << endl;


	//(8)显示图片

	imshow("img", img);

	waitKey(0);


	return 0;

}