Unverified Commit 28b91429 authored by pkulzc's avatar pkulzc Committed by GitHub
Browse files

Merge pull request #4147 - update exporting_models doc

Object detection export dir
parents c0a380d2 a4220162
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
After your model has been trained, you should export it to a Tensorflow After your model has been trained, you should export it to a Tensorflow
graph proto. A checkpoint will typically consist of three files: graph proto. A checkpoint will typically consist of three files:
* model.ckpt-${CHECKPOINT_NUMBER}.data-00000-of-00001, * model.ckpt-${CHECKPOINT_NUMBER}.data-00000-of-00001
* model.ckpt-${CHECKPOINT_NUMBER}.index * model.ckpt-${CHECKPOINT_NUMBER}.index
* model.ckpt-${CHECKPOINT_NUMBER}.meta * model.ckpt-${CHECKPOINT_NUMBER}.meta
...@@ -16,7 +16,12 @@ python object_detection/export_inference_graph.py \ ...@@ -16,7 +16,12 @@ python object_detection/export_inference_graph.py \
--input_type image_tensor \ --input_type image_tensor \
--pipeline_config_path ${PIPELINE_CONFIG_PATH} \ --pipeline_config_path ${PIPELINE_CONFIG_PATH} \
--trained_checkpoint_prefix ${TRAIN_PATH} \ --trained_checkpoint_prefix ${TRAIN_PATH} \
--output_directory output_inference_graph.pb --output_directory ${EXPORT_DIR}
``` ```
Afterwards, you should see a graph named output_inference_graph.pb. Afterwards, you should see the directory ${EXPORT_DIR} containing the following:
* output_inference_graph.pb, the frozen graph format of the exported model
* saved_model/, a directory containing the saved model format of the exported model
* model.ckpt.*, the model checkpoints used for exporting
* checkpoint, a file specifying to restore included checkpoint files
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment