Skip to content

Instantly share code, notes, and snippets.

@tomekdz

tomekdz/gist:b16c5d5f649cf2bb363719e13f8aa0cb

Created September 20, 2018 12:29
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 tomekdz/b16c5d5f649cf2bb363719e13f8aa0cb to your computer and use it in GitHub Desktop.
Save tomekdz/b16c5d5f649cf2bb363719e13f8aa0cb to your computer and use it in GitHub Desktop.
Download ZIP
Converting tensorflow frozen graph to .tflite
Raw
gistfile1.txt
./bazel-bin/tensorflow/contrib/lite/toco/toco \
--input_file=yolov2-tiny.pb \
--output_file=foo.tflite  \
--input_format=TENSORFLOW_GRAPHDEF \
--output_format=TFLITE \
--input_arrays=input \
--input_shapes=1,416,416,3 \
--output_arrays=output
@iseegr8tfuldeadppl
Copy link

if toco: error: one of the arguments --graph_def_file --saved_model_dir --keras_model_file is required
error shows up make sure you replace input_file with graph_def_file

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment