命令行使用模型

python  \
./tensorflow/examples/label_image/label_image.py  \
--graph=/tmp/mobilenet_v2_224_graph.pb  \
--input_layer=Placeholder  \
--output_layer=final_result  \
--labels=/tmp/mobilenet_v2_224_labels.txt \
--image=/home/test/data/test/Cat_Meow_2-Cat_Stevens-2034822903.jpg  

如果不指定--input_layer=Placeholder,会报如下错误

The name 'import/input' refers to an Operation not in the graph.

由于新的模型input_layer名称变为了Placeholder,需要指定来解决这个报错

如果不指定--output_layer=final_result, 会有如下报错

KeyError: "The name 'import/InceptionV3/Predictions/Reshape_1' 
refers to an Operation not in the graph."

如果不指定label, 会报找不到label文件的错

tensorflow.python.framework.errors_impl.NotFoundError: 
tensorflow/examples/label_image/data/imagenet_slim_labels.txt; 
No such file or directory

你可能感兴趣的:(命令行使用模型)