how to convert .ckpt file to .pb

后端 未结 2 1225
孤独总比滥情好
孤独总比滥情好 2021-01-06 14:37

I use ssd_mobilenets in Object detection API to train my own model, and get .ckpt files. It works well on my computer, but now I want to use the model on my phone. So, I nee

2条回答
  •  南方客
    南方客 (楼主)
    2021-01-06 15:05

    Use export_inference_graph.py to convert model checkpoint file into a .pb file.

    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/
    

提交回复
热议问题