0

我正在关注Tensorflow 以获取重新训练模型的诗人指导。我已成功创建 retrained_graph.pb 和 retrained_labels.txt。虽然我使用 imagenet_comp_graph_label_strings.txt 和 tensorflow_inception_graph.pb 然后应用程序运行没有任何错误。但是使用我创建的文件然后我得到错误:

原因:java.lang.UnsupportedOperationException: Op BatchNormWithGlobalNormalization 在 GraphDef 版本 21 中不可用。它已在版本 9 中删除。使用 tf.nn.batch_normalization()。在 org.tensorflow.Graph.importGraphDef(Native Method) 在 org.tensorflow.Graph.importGraphDef(Graph.java:118) 在 org.tensorflow.Graph.importGraphDef(Graph.java:102) 在 org.tensorflow.contrib.android .TensorFlowInferenceInterface.load(TensorFlowInferenceInterface.java:402) 在 org.tensorflow.contrib.android.TensorFlowInferenceInterface.initializeTensorFlow(TensorFlowInferenceInterface.java:91) 在 org.tensorflow.demo.TensorFlowImageClass

我还尝试按照TensorFlow 指南从头开始重新训练模型,但我得到了同样的错误。我还尝试剥离图表并更改 ClassifierActivity。但我仍然得到同样的错误。

4

1 回答 1

0

看起来您正在使用两个不兼容的 TensorFlow 版本,一个用于训练第一个模型,另一个是从 java 调用的旧模型。

您如何使用创建的文件?您可以重建执行此操作的程序以使用较新版本的 tf 吗?

于 2017-03-15T17:53:28.537 回答