获取某一层权重

ref

方法一:通过get_layer()函数先获取要获取权重对应的层;接着通过get_weights()

函数获取改成对应的全部参数(是一个长的为2的列表,列表中的每一个元素是一个numpy数组,第一存放该层的权重W的参数,第一个存放偏置b)
layer的两个函数:
get_weights(), set_weights(weights)

model = load_model('vgg.h5')
layer1 = model.get_layer(index=2)
weights = layer1.get_weights()   #获取该层的参数W和b

通过在pycharm中调试运行,查看model的具体内容,进一步打开layers变量,可以查看到我们要获取层的索引。在本例子中,要获取的索引为2,


image.png

方法二:通过model.get_weights()先获取模型的全部参数(一个列表数组,第一层W,第一层b,第二层W,第二层b,...)

weights = model.get_weights() #获取整个网络模型的全部参数
print(weights [0].shape)  #第一层的w
print(weights [1].shape)  #第一层的b
print(weights [2].shape)  #第二层的w
print(weights [3].shape)  #第二层的b
image.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容