Skip to content

Instantly share code, notes, and snippets.

@Namburger

Namburger/export_inference_graph.py

Created August 15, 2020 01:31
Show Gist options
  • Save Namburger/74dbb145e7220551e48ed55278b78e46 to your computer and use it in GitHub Desktop.
Save Namburger/74dbb145e7220551e48ed55278b78e46 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
export_inference_graph.py
# Make inference graph.
!python3 /content/models/research/object_detection/export_inference_graph.py \
--input_type=image_tensor \
--pipeline_config_path=/content/models/research/object_detection/samples/configs/ssdlite_mobiledet_edgetpu_320x320_coco_sync_4x4.config \
--output_directory=/content/inference_graph \
--trained_checkpoint_prefix=/content/train/model.ckpt-25000 # Make sure to change this checkpoint to the corresponding num step you set from above.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment