Skip to content

Instantly share code, notes, and snippets.

@bmabir17

bmabir17/convert_save.py

Last active August 13, 2020 16:45
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 bmabir17/5d8703ad10f7fc2f74cfa5d7d69a7943 to your computer and use it in GitHub Desktop.
Save bmabir17/5d8703ad10f7fc2f74cfa5d7d69a7943 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
convert_save.py
# Default Optimization is choosen
converter.optimizations = [tf.lite.Optimize.DEFAULT]
# Convert to TFLite Model
tflite_model = converter.convert()
# Save Model as tflite format
tflite_path = "deeplabv3_mnv2_custom_257.tflite"
tflite_model_size = open(tflite_path, 'wb').write(tflite_model)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment