keras 权重保存与读取，迁移指定层参数，模型layer改名

最新推荐文章于 2024-06-24 19:42:48 发布

沃·夏澈德

最新推荐文章于 2024-06-24 19:42:48 发布

阅读量1.1k

点赞数 3

分类专栏： keras相关文章标签： keras 模型迁移迁移学习

本文链接：https://blog.csdn.net/aaalswaaa1/article/details/120468718

版权

模型权重参数迁移层名改名模型实例权重加载

关键词由CSDN通过智能技术生成

keras相关专栏收录该内容

1 篇文章 0 订阅

订阅专栏

weight_Dense_1,bias_Dense_1=model.get_layer('dense').get_weights()
weight_Dense_2,bias_Dense_2=model2.get_layer('dense').get_weights()

weight_conv_1,bias_conv_1=model.get_layer('conv1').get_weights()
weight_conv_2,bias_conv_2=model2.get_layer('conv1').get_weights()

model2.get_layer('conv1')._name = 'hah'


model2.load_weights('./demo.h5',by_name=True)

weight_Dense_1a,bias_Dense_1a=model.get_layer('dense').get_weights()
weight_Dense_2a,bias_Dense_2a=model2.get_layer('dense').get_weights()

weight_conv_1a,bias_conv_1a=model.get_layer('conv1').get_weights()
weight_conv_2a,bias_conv_2a=model2.get_layer('hah').get_weights()

其中，model，model2是两个同模型的不同实例，模型有默认命名，所以需要改名，不然所有同名的层的参数都会被读取到model2.