Skip to content

Instantly share code, notes, and snippets.

@louis030195
Created September 5, 2019 13:08
Show Gist options
  • Save louis030195/a66677db4b690a46dccfae06a797d5f6 to your computer and use it in GitHub Desktop.
Save louis030195/a66677db4b690a46dccfae06a797d5f6 to your computer and use it in GitHub Desktop.
# Vector shape (-1, -1, -1, 3) = image_tensor
# Base64 string shape (-1) = encoded_image_string_tensor
INPUT_TYPE = 'encoded_image_string_tensor'
python models/research/object_detection/export_inference_graph.py \
--input_type $INPUT_TYPE \
--pipeline_config_path model/pipeline.config \
--trained_checkpoint_prefix model/model.ckpt \
--inference_graph_path output_inference_graph.pb \
--output_directory exported_model
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment