我們用keras訓練模型后,通常保存的模型格式類型為hdf5格式,也就是.h5文件。
但如果我們想要移植到移動端,特別是基於tensorflow支持的移動端,那就需要轉換成tflite格式。
如何轉換呢?在tensorflow1.9及以上版本,支持通過命令行方式方便進行轉換:
tflite_convert --output_file=/home/yourname/Documents/tensorflow/images/500pics2/model/my_model.tflite --keras_model_file=/home/yourname/Documents/tensorflow/images/500pics2/model/my_model.h5