Skip to content

Instantly share code, notes, and snippets.

@universax

universax/freeze_graph.sh

Created February 14, 2019 02:45
Show Gist options
  • Save universax/9b4433138f1cda92d1fb1540beacba51 to your computer and use it in GitHub Desktop.
Save universax/9b4433138f1cda92d1fb1540beacba51 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
freeze_graph.sh
freeze_graph \
--input_graph="/home/user_name/output/mobilenet_v1_mymodel.pb" \
--input_binary=true \
--input_checkpoint="/home/user_name/train/model.ckpt-1000" \
--output_graph="/home/user_name/output/mobilenet_v1_mymodel_frozen.pb" \
--output_node_names=MobilenetV1/Predictions/Reshape_1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment