如何用自己的数据集重新创建 .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 争论）。所以你不用担心这个。