Skip to content

Instantly share code, notes, and snippets.

@Namburger

Namburger/export_tflite_graph.py

Created August 15, 2020 02:08
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Namburger/5f15b35479a8818ad02ad2fcf4d216a8 to your computer and use it in GitHub Desktop.
Save Namburger/5f15b35479a8818ad02ad2fcf4d216a8 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
export_tflite_graph.py
# Now we export this model to tflite_graph format.
%cd /content/models/research
!mkdir /content/output_model
!python3 object_detection/export_tflite_ssd_graph.py \
--pipeline_config_path=/content/models/research/object_detection/samples/configs/ssdlite_mobiledet_edgetpu_320x320_coco_sync_4x4.config \
--trained_checkpoint_prefix=/content/train/model.ckpt-25000 \
--output_directory=/content/output_model \
--add_postprocessing_op=true
# Make sure to change the model-ckpt-# to match the checkpoint number you used.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment