ML之模型文件:机器学习、深度学习中常见的模型文件(.h5、.keras)简介、h5模型文件下载集锦、使用方法之详细攻略
生活随笔
收集整理的這篇文章主要介紹了
ML之模型文件:机器学习、深度学习中常见的模型文件(.h5、.keras)简介、h5模型文件下载集锦、使用方法之详细攻略
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
ML之模型文件:機器學習、深度學習中常見的模型文件(.h5、.keras)簡介、h5模型文件下載集錦、使用方法之詳細攻略
?
?
?
目錄
ML/DL中常見的模型文件(.h5、.keras)簡介及其使用方法
一、.h5文件
1、常見的h5文件下載
二、.keras文件
1、模型的保存和載入
?
?
?
?
?
?
ML/DL中常見的模型文件(.h5、.keras)簡介及其使用方法
一、.h5文件
可使用model.save(filepath)函數,將Keras模型和權重保存在一個HDF5文件中,h5文件將包含:
- 模型的結構,以便重構該模型
- 模型的權重
- 訓練配置(損失函數,優化器等)
- 優化器的狀態,以便于從上次訓練中斷的地方開始
(1)、模型的保存和載入
model_path = 'model.h5' model.save(model_path ) '保存模型'from keras.models import load_model model = load_model(model_path ) '載入模型'model_weights_path = 'model_weights.h5' model.save_weights(model_weights_path ) '保存模型的權重' model.load_weights(model_weights_path ) '載入模型的權重''如果你需要加載權重到不同的網絡結構(有些層一樣)中,例如fine-tune或transfer-learning,你可以通過層名字來加載模型' model.load_weights('model_weights.h5', by_name=True)1、常見的h5文件下載
resnet50_coco_best_v2.1.0.h5模型文件
下載地址:https://github.com/fizyr/keras-retinanet/releases
?
?
?
?
二、.keras文件
1、模型的保存和載入
'保存模型' path_model = 'Functional_model.keras' # Functional model2.save(path_model) #保存訓練好的Keras模型的權重只需要調用一個函數 del model2 #當不再需要用到這個模型時可以從內存中刪除它。'加載模型' path_model = 'Functional_model.keras' from tensorflow.python.keras.models import load_model model3 = load_model(path_model)?
?
?
?
?
?
?
《新程序員》:云原生和全面數字化實踐50位技術專家共同創作,文字、視頻、音頻交互閱讀總結
以上是生活随笔為你收集整理的ML之模型文件:机器学习、深度学习中常见的模型文件(.h5、.keras)简介、h5模型文件下载集锦、使用方法之详细攻略的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: DL之CNN:利用卷积神经网络算法(2→
- 下一篇: 成功解决WARNING:tensorfl