convert keras model into tflite model

9996bdf3 · MithilaGunasinghe · 246f0ea4 · 9996bdf3
Commit 9996bdf3 authored Oct 19, 2020 by MithilaGunasinghe
Hide whitespace changes
Inline Side-by-side

Showing with 12 additions and 0 deletions

CNNmodel/convert.py CNNmodel/convert.py +12 -0

No files found.
--- a/CNNmodel/convert.py
+++ b/CNNmodel/convert.py
+'''
+Author : Gunasinghe M.D.
+IT NUM : IT17043342
+
+'''
+
+import tensorflow as tf
+
+
+converter = tf.lite.TFLiteConverter.from_keras_model_file('shapes_classification.h5') 
+tfmodel = converter.convert() 
+open ("model.tflite" , "wb") .write(tfmodel)
\ No newline at end of file