How to convert .ckpt to .pb?

前端 未结 3 905
渐次进展
渐次进展 2020-12-30 13:24

I am new to deep learning and I want to use a pretrained (EAST) model to serve from the AI Platform Serving, I have these files made available by the developer:

3条回答
  •  爱一瞬间的悲伤
    2020-12-30 14:01

    If you specify INPUT_TYPE as image_tensor and PIPELINE_CONFIG_PATH as your config file with this command.

    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}
    

    you can get your model in 3 formats in your export dir;

    • frozen_graph.pb
    • savedmodel.pb
    • checkpoint

    for more info https://github.com/tensorflow/models/tree/master/research/object_detection

提交回复
热议问题