如何用自己的數據集重新創建.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參數來更改它）。所以你不需要擔心這一點。