keras读取h5文件load_weights、load的操作方法-创新互联

不懂keras读取h5文件load_weights、load的操作方法？其实想解决这个问题也不难，下面让小编带着大家一起学习怎么去解决，希望大家阅读完这篇文章后大所收获。

关于保存h6模型、权重网上的示例非常多，也非常简单。主要有以下两个函数：

1、keras.models.load_model() 读取网络、权重

2、keras.models.load_weights() 仅读取权重

load_model代码包含load_weights的代码，区别在于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') 

另外有需要云服务器可以了解下创新互联scvps.cn，海内外云服务器15元起步，三天无理由+7*72小时售后在线，公司持有idc许可证，提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案，具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势，专为企业上云打造定制，能够满足用户丰富、多元化的应用场景需求。