Skip to content

Instantly share code, notes, and snippets.

@karimamd

karimamd/graph pb to tflite (Command line)

Created June 22, 2019 18: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 karimamd/4b5521566ab851bb5210cb81616ec639 to your computer and use it in GitHub Desktop.
Save karimamd/4b5521566ab851bb5210cb81616ec639 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
graph pb to tflite (Command line)
!toco \
--graph_def_file=output_graph.pb \
--output_file=xd_graph.tflite \
--output_format=TFLITE \
--input_shape=32,256,192,3 \
--input_array= NAME OF INPUT MODEL OF YOUR MODEL \
--output_array= NAME OF OUTPUT LAYER OF YOUR MODEL \
--inference_type=FLOAT \
--inference_input_type=FLOAT \
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment