电脑训练的retrained_graph.pb需转化，放到Tesorflow Lite上

今天将电脑训练出来的retrained_graph.pb文件，放到Tesorflow Lite上，报

java.lang.RuntimeException: Node ‘MobilenetV1/Predictions/Softmax’ does not exist in model ‘file:///android_asset/graph.pb’

at org.tensorflow.contrib.android.TensorFlowInferenceInterface.graphOperation(TensorFlowInferenceInterface.java:171)

at org.tensorflow.demo.TensorFlowImageClassifier.create(TensorFlowImageClassifier.java:106)

错误。

查了下资料，电脑训练的retrained_graph.pb需转化，放到Tesorflow Lite上。

可查考

http://nilhcem.com/android/custom-tensorflow-classifier

python tensorflow/python/tools/optimize_for_inference.py \

–input=/tf_files/retrained_graph.pb \

–output=/tf_files/optimized_graph.pb \

–input_names=”Mul” \

–output_names=”final_result”

2018年1月28日