How to convert keras(h5) file to a tflite file?

by 
from tensorflow.contrib import lite
converter = lite.TFLiteConverter.from_keras_model_file( 'model.h5')
tfmodel = converter.convert()
open ("model.tflite" , "wb") .write(tfmodel)

You can use the TFLiteConverter to directly convert .h5 files to .tflite file.
This does not work on Windows.

For Windows, use this Google Colab notebook to convert. Upload the .h5 file and it will convert it .tflite file.

Follow, if you want to try it yourself :

  1. Create a Google Colab Notebook. In the left top corner, click the “UPLOAD” button and upload your .h5 file.

  2. Create a code cell and insert this code.

    from tensorflow.contrib import lite
converter = lite.TFLiteConverter.from_keras_model_file( 'model.h5' ) # Your model's name
model = converter.convert()
file = open( 'model.tflite' , 'wb' ) 
file.write( model )

  3. Run the cell. You will get a model.tflite file. Right click on the file and select “DOWNLOAD” option.

Leave a Comment