我是张力流的新手，我正在关注诗人2教程的张量流并克隆他们的回购：https://github.com/googlecodelabs/tensorflow-for-poets-2
在本教程中，我学习了如何运行脚本进行重新训练，评估，优化推理，在cmd中使用quantize_graph。 现在我对“冻结”感到好奇。
在这个stackoverflow问题中，我学习了如何在cmd中运行freeze_graph： How to use freeze_graph.py tool in TensorFlow v1
问题是这个所谓的“检查点”，我在遵循诗人2的教程张量流时没有。我的问题是如何获得此检查点文件？由于我在训练数据后只有.pb和标签
注意：我知道上面的repo中没有freeze_graph.py，这就是为什么我从https://github.com/tensorflow/tensorflow/tree/master/tensorflow/python/tools复制它，希望我可以使用它。
由于
答案 0 :(得分：0)
要获取检查点文件，我使用了瘦库中的脚本train_image_classifier.py：https://github.com/tensorflow/models/tree/master/research/slim#Training。
它在我所做的不同测试中运行良好，也适用于新的数据集。您可以从头开始训练或微调现有模型。
此致 斯蒂芬
答案 1 :(得分：0)
.pb文件是检查点文件的冻结图。您需要的是重新训练您的模型并将其保存为TF检查点文件格式。使用tf.train.Saver执行此操作。有关其用法，请参阅this instruction。然后你可以使用freeze_graph.py做你想做的事。