Tensorflow模型在Python中工作,但在C+中不工作+; 对于一个小的背景,我的主要目标是使用Tensorflow的C++ API对图像进行分类,并将时间划分到不同的系统上。

Tensorflow模型在Python中工作,但在C+中不工作+; 对于一个小的背景,我的主要目标是使用Tensorflow的C++ API对图像进行分类,并将时间划分到不同的系统上。,python,c++,tensorflow,Python,C++,Tensorflow,我曾经将他的Caffe模型转换为Tensorflow,一旦打开它,就会生成vgg16.tfmodel文件,它看起来是一个.pb文件 使用Ry的tf_forward.py来运行这个结果文件似乎效果很好,可以对猫、狗等进行分类。但是,当我修改label_图像示例(tensorflow/examples/label_image/)以使用新的vgg16.pb文件时,似乎出现了一些问题 下面是对tensorflow-vgg16示例中的cat图片进行分类的输出: I tensorflow/examples/

我曾经将他的Caffe模型转换为Tensorflow,一旦打开它,就会生成vgg16.tfmodel文件,它看起来是一个.pb文件

使用Ry的tf_forward.py来运行这个结果文件似乎效果很好,可以对猫、狗等进行分类。但是,当我修改label_图像示例(tensorflow/examples/label_image/)以使用新的vgg16.pb文件时,似乎出现了一些问题

下面是对tensorflow-vgg16示例中的cat图片进行分类的输出:

I tensorflow/examples/label_image/main.cc:195] n03729826 matchstick (644): 0.0762781
I tensorflow/examples/label_image/main.cc:195] n01930112 nematode, nematode worm, roundworm (111): 0.0493599
I tensorflow/examples/label_image/main.cc:195] n03196217 digital clock (530): 0.0317698
I tensorflow/examples/label_image/main.cc:195] n03666591 lighter, light, igniter, ignitor (626): 0.0313151
I tensorflow/examples/label_image/main.cc:195] n04286575 spotlight, spot (818): 0.0252088
以下是我在修改label_image/main.cc时使用的设置:

string image = "tensorflow/examples/label_image/data/cat.jpg"; //from Ry's project
string graph = "tensorflow/examples/label_image/data/vgg16.pb"; //renamed from Ry's project
string labels ="tensorflow/examples/label_image/data/synset.txt"; //from Ry's project
int32 input_width = 224;
int32 input_height = 224;
int32 input_mean = 128;
int32 input_std = 128;
string input_layer = "mul";
string output_layer = "prob";
bool self_test = false;
string root_dir = "";

我也尝试过使用freeze_graph.py来获取模型和权重,但始终无法使其正常工作。在对所有内容进行了追溯之后,我遇到了protobuf代码,有点卡住了。任何帮助都将不胜感激:)

对于将来关注此问题的人,此问题是由于使用了错误的输入层名称造成的