Skip to content

GitLab

Commit 3141ea44 authored by prats226's avatar prats226 Committed by GitHub
Browse files

Update running_pets.md 

Updated documentation. Command for exporting model checkpoint to frozen graph.
parent 78bb025c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
object_detection/g3doc/running_pets.md
View file @ 3141ea44
...@@ -296,8 +296,8 @@ gsutil cp gs://${YOUR_GCS_BUCKET}/train/model.ckpt-${CHECKPOINT_NUMBER}.* . ...@@ -296,8 +296,8 @@ gsutil cp gs://${YOUR_GCS_BUCKET}/train/model.ckpt-${CHECKPOINT_NUMBER}.* .
python object_detection/export_inference_graph.py \ python object_detection/export_inference_graph.py \
--input_type image_tensor \ --input_type image_tensor \
--pipeline_config_path object_detection/samples/configs/faster_rcnn_resnet101_pets.config \ --pipeline_config_path object_detection/samples/configs/faster_rcnn_resnet101_pets.config \
--checkpoint_path model.ckpt-${CHECKPOINT_NUMBER} \ --trained_checkpoint_prefix model.ckpt-${CHECKPOINT_NUMBER} \
--inference_graph_path output_inference_graph.pb --output_directory output_inference_graph.pb
``` ```
Afterwards, you should see a graph named `output_inference_graph.pb`. Afterwards, you should see a graph named `output_inference_graph.pb`.
......
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