tensorflow - 如何将 .ckpt 文件转换为 .pb

Question

我在对象检测 API 中使用 ssd_mobilenets 来训练我自己的模型，并获取 .ckpt 文件。它在我的电脑上运行良好，但现在我想在我的手机上使用该模型。所以，我需要将它转换为 .pb 文件。我不知道该怎么做，有人可以帮忙吗？顺便说一下，ssd_mobilenets 的图表很复杂，我找不到哪个是模型的输出。有没有人知道输出的名称？

Answer 1

使用export_inference_graph.py将模型检查点文件转换为 .pb 文件。

python tensorflow_models/object_detection/export_inference_graph.py \
--input_type image_tensor \
--pipeline_config_path architecture_used_while_training.config \
--trained path_to_saved_ckpt/model.ckpt-NUMBER \
--output_directory model/