我是深度学习的新手,我想使用预训练 (EAST) 模型从 AI Platform Serving 提供服务,开发人员提供了以下文件:
- model.ckpt-49491.data-00000-of-00001
- 检查点
- 模型.ckpt-49491.index
- 型号.ckpt-49491.meta
我想把它转换成 TensorFlow .pb
格式。有办法吗?我从 这里 拿走了模型
完整代码可 在此处 获得。
我在 这里 查找,它显示了以下代码来转换它:
来自 tensorflow/models/research/
INPUT_TYPE=image_tensor
PIPELINE_CONFIG_PATH={path to pipeline config file}
TRAINED_CKPT_PREFIX={path to model.ckpt}
EXPORT_DIR={path to folder that will be used for export}
python object_detection/export_inference_graph.py \
--input_type=${INPUT_TYPE} \
--pipeline_config_path=${PIPELINE_CONFIG_PATH} \
--trained_checkpoint_prefix=${TRAINED_CKPT_PREFIX} \
--output_directory=${EXPORT_DIR}
我无法弄清楚要传递什么值:
- 输入类型
- PIPELINE_CONFIG_PATH。
原文由 Shivam Sahu 发布,翻译遵循 CC BY-SA 4.0 许可协议
这是将检查点转换为 SavedModel 的代码