TensorFlow 对象检测 API 教程 - 第5部分:保存和部署模型
在本教程的这一步,认为已经选择了预先训练的对象检测模型,调整现有的数据集或创建自己的数据集,并将其转换为 TFRecord
文件,修改模型配置文件并开始训练。但是,现在需要保存模型并将其部署到项目中。
一. 将检查点模型 (.ckpt)
保存为 .pb
文件
回到 TensorFlow
对象检测文件夹,并将 export_inference_graph.py
文件复制到包含模型配置文件的文件夹中。
python export_inference_graph.py --input_type image_tensor --pipeline_config_path