建站
咨詢
售后
建站咨詢
新聞中心
這里有您想知道的互聯(lián)網(wǎng)營銷解決方案
keras讀取h5文件load_weights、load的操作方法-創(chuàng)新互聯(lián)
不懂keras讀取h5文件load_weights、load的操作方法？其實想解決這個問題也不難，下面讓小編帶著大家一起學習怎么去解決，希望大家閱讀完這篇文章后大所收獲。
關(guān)于保存h6模型、權(quán)重網(wǎng)上的示例非常多，也非常簡單。主要有以下兩個函數(shù)：
1、keras.models.load_model() 讀取網(wǎng)絡、權(quán)重
2、keras.models.load_weights() 僅讀取權(quán)重
load_model代碼包含load_weights的代碼，區(qū)別在于load_weights時需要先有網(wǎng)絡、并且load_weights需要將權(quán)重數(shù)據(jù)寫入到對應網(wǎng)絡層的tensor中。
下面以resnet50加載h6權(quán)重為例，示例代碼如下
import keras from keras.preprocessing import image import numpy as np from network.resnet50 import ResNet50 #修改過，不加載權(quán)重（默認官方加載亦可） model = ResNet50() # 參數(shù)默認 by_name = Fasle， 否則只讀取匹配的權(quán)重 # 這里h6的層和權(quán)重文件中層名是對應的（除input層） model.load_weights(r'\models\resnet50_weights_tf_dim_ordering_tf_kernels_v2.h6')
另外有需要云服務器可以了解下創(chuàng)新互聯(lián)scvps.cn，海內(nèi)外云服務器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應用場景需求。
名稱欄目：keras讀取h5文件load_weights、load的操作方法-創(chuàng)新互聯(lián)
文章網(wǎng)址：http://fisionsoft.com.cn/article/psjhh.html
