电脑训练的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日