當前位置:
首頁 >
DeepLearning:模型之间的相互转化(keras-hdf5→Tensorflow-pb文件)
發(fā)布時間:2025/3/20
53
豆豆
生活随笔
收集整理的這篇文章主要介紹了
DeepLearning:模型之间的相互转化(keras-hdf5→Tensorflow-pb文件)
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
【寫在前面】
在深度學習的應用中,我們往往需要將Python中訓練好的模型文件部署到實際的應用測試中,所以自然而然我們就需要進行模型之間的相互轉換,例如本人在Python環(huán)境下訓練好的hdf5文件,現(xiàn)在要加載到C++的Tensorflow部署環(huán)境完成實際的測試工作。
【轉換步驟】
1、獲取keras的hdf5模型
2、選擇樣本完成keras的hdf5模型測試,記錄測試結果
3、通過代碼實現(xiàn)hdf5模型到tensorflow pb模型的轉換
4、再次輸入樣本,完成pb模型測試,并將測試結果和hdf5模型的測試結果進行比較,評價模型轉換是否成功。
【轉換實戰(zhàn)】
1.hdf5模型的生成和測試大家根據(jù)自己的訓練過程和結果完成;
2.hdf5模型的轉化;
【轉換測試】
skimage庫安裝:
conda install scikit-image
總結
以上是生活随笔為你收集整理的DeepLearning:模型之间的相互转化(keras-hdf5→Tensorflow-pb文件)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Python:tkinter滚动抽奖器
- 下一篇: 【图像处理】纹理检测算法