maifeeulasad/converter.py

Created April 21, 2021 11:30

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/maifeeulasad/d7e67b50e378584b9f3a9761eaa93da6.js"></script>
Save maifeeulasad/d7e67b50e378584b9f3a9761eaa93da6 to your computer and use it in GitHub Desktop.

Download ZIP

Revisions

maifeeulasad created this gist Apr 21, 2021.

converter.py

-Original file line number
+Diff line change
@@ -0,0 +1,7 @@
+    import tensorflow as tf
+    model = tf.keras.models.load_model('name_gender.h5')
+    converter = tf.lite.TFLiteConverter.from_keras_model(model)
+    tflite_model = converter.convert()
+    open("converted_model.tflite", "wb").write(tflite_model)