Skip to content

Instantly share code, notes, and snippets.

@haoxi911

haoxi911/tflite.sh

Created May 8, 2018 05:35
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 haoxi911/c00d27c2aab15e79d4bb822bf44d1573 to your computer and use it in GitHub Desktop.
Save haoxi911/c00d27c2aab15e79d4bb822bf44d1573 to your computer and use it in GitHub Desktop.
Download ZIP
Script to convert TensorFlow GraphDef to TensorFlow Lite
Raw
tflite.sh
bazel-bin/tensorflow/contrib/lite/toco/toco \
--input_file=/home/ubuntu/pps_output/output_graph.pb \
--output_file=/home/ubuntu/pps_output/output_graph.tflite \
--input_arrays=Placeholder \
--output_arrays=final_result \
--inference_type=FLOAT \
--input_shapes=1,224,224,3 \
--input_format=TENSORFLOW_GRAPHDEF \
--output_format=TFLITE
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment