使用带有classify_image.py的重新训练数据

Question

我一直在使用张量流图像识别。我已经构建了许多与classify_image.py交互的脚本。

我还使用retrain.py和我自己的数据集重新训练了模型。

如何使用生成的两个文件：output_graph.pb，output_labels.txt with classify_image.py？

Answer 1

啊，the docs say

  

如果您想在Python程序this example from @eldor4do shows what you'll need to do.

中使用重新训练的模型

在本地复制/编辑了一个文件，python .\edited-retraining-example.py。这很容易。

请注意，如果您使用的是Windows，请将/tmp/...的所有示例更改为c:/tmp/...。