如何将基于tensorflow构建的修改模型（用于转移学习，格式为.ckpt）导出到冻结图（.pb文件）

Question

我正在尝试从argman/EAST导出此特定模型的导出。模型是单独构建的，需要时会在会话中加载权重。是否有一种方法可以将模型从会话导出到冻结图（.pb）文件。还是有其他方法可以做到这一点？模型在model.py file(line 33)中构建，权重在run_demo_server.py (line 54-57)中加载到模型

我已经看到了一些将“.ckpt”文件转换为“.pb”的脚本，但是，所有这些脚本都要求输出节点名称，我无法找到输出层名称。感谢您的帮助

提前谢谢

Answer 1

在google colab中试试这个

!python export_inference_graph.py  input_type image_tensor  pipeline_config_path ssd_mobilenet_v2_quantized_300x300_coco.config  trained_checkpoint_prefix training/model.ckpt-xxxx  output_directory inference_graph

还是这个在cmd里

python export_inference_graph.py  input_type image_tensor  pipeline_config_path training/faster_rcnn_inception_v2_pets.config  trained_checkpoint_prefix training/model.ckpt-XXXX  output_directory inference_graph