不懂keras讀取h5文件load_weights、load的操作方法?其實想解決這個問題也不難,下面讓小編帶著大家一起學習怎么去解決,希望大家閱讀完這篇文章后大所收獲。
創(chuàng)新互聯公司-專業(yè)網站定制、快速模板網站建設、高性價比贛縣網站開發(fā)、企業(yè)建站全套包干低至880元,成熟完善的模板庫,直接使用。一站式贛縣網站制作公司更省心,省錢,快速模板網站建設找我們,業(yè)務覆蓋贛縣地區(qū)。費用合理售后完善,十余年實體公司更值得信賴。關于保存h6模型、權重網上的示例非常多,也非常簡單。主要有以下兩個函數:
1、keras.models.load_model() 讀取網絡、權重
2、keras.models.load_weights() 僅讀取權重
load_model代碼包含load_weights的代碼,區(qū)別在于load_weights時需要先有網絡、并且load_weights需要將權重數據寫入到對應網絡層的tensor中。
下面以resnet50加載h6權重為例,示例代碼如下
import keras from keras.preprocessing import image import numpy as np from network.resnet50 import ResNet50 #修改過,不加載權重(默認官方加載亦可) model = ResNet50() # 參數默認 by_name = Fasle, 否則只讀取匹配的權重 # 這里h6的層和權重文件中層名是對應的(除input層) model.load_weights(r'\models\resnet50_weights_tf_dim_ordering_tf_kernels_v2.h6')
另外有需要云服務器可以了解下創(chuàng)新互聯scvps.cn,海內外云服務器15元起步,三天無理由+7*72小時售后在線,公司持有idc許可證,提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業(yè)上云的綜合解決方案,具有“安全穩(wěn)定、簡單易用、服務可用性高、性價比高”等特點與優(yōu)勢,專為企業(yè)上云打造定制,能夠滿足用戶豐富、多元化的應用場景需求。