如何將hdf5文件轉換成tflite文件


我們用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

其他方式可以參見 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lite/toco/g3doc/cmdline_examples.md#keras

 

  


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM