方法一:通過get_layer()函數(shù)先獲取要獲取權(quán)重對應(yīng)的層伙菊;接著通過get_weights()
函數(shù)獲取改成對應(yīng)的全部參數(shù)(是一個長的為2的列表降铸,列表中的每一個元素是一個numpy數(shù)組贝乎,第一存放該層的權(quán)重W的參數(shù)啃匿,第一個存放偏置b)
layer的兩個函數(shù):
get_weights(), set_weights(weights)
model = load_model('vgg.h5')
layer1 = model.get_layer(index=2)
weights = layer1.get_weights() #獲取該層的參數(shù)W和b
通過在pycharm中調(diào)試運行晃择,查看model的具體內(nèi)容握童,進一步打開layers變量姆怪,可以查看到我們要獲取層的索引。在本例子中澡绩,要獲取的索引為2稽揭,
image.png
方法二:通過model.get_weights()先獲取模型的全部參數(shù)(一個列表數(shù)組,第一層W肥卡,第一層b溪掀,第二層W,第二層b步鉴,...)
weights = model.get_weights() #獲取整個網(wǎng)絡(luò)模型的全部參數(shù)
print(weights [0].shape) #第一層的w
print(weights [1].shape) #第一層的b
print(weights [2].shape) #第二層的w
print(weights [3].shape) #第二層的b
image.png