如何使用自己的数据集重新创建.pb文件

Question

我正在使用tensorflow和Inception v3。 如果我使用自己的数据集进行培训，如何重新创建retrain.py中使用的文件（classify_image_graph_def.pb）？

Answer 1

当你在自己的数据集上运行retrain.py时，它会自动在/ tmp目录中创建output_graph.pb和output_labels.txt（你可以通过传递--output_graph参数来改变它）。所以你不用担心这个。