读取.h5里面的数据

本文介绍了如何读取.h5文件中存储的神经网络参数。作者分享了用于读取含有五个卷积层参数的.h5文件的代码,这对于利用预训练模型进行迁移学习非常有用。
摘要由CSDN通过智能技术生成

我想要利用别人的训练过的参数来训练自己的神经网络,我遇到两种,一种参数是在mat文件里,一种是在.h5文件里。mat文件数据很容易读取。下面给出读取.h5文件里储存数据的代码:

import h5py
f = h5py.File('./pretrained/weights_c3d.h5','r')
for key, value in f.attrs.items():
    print("  {}: {}".format(key, value))
            

group1 = f.get('conv3d_1')
group2 = f.get('conv3d_2')
group3 = f.get('conv3d_3')
group4 = f.get('conv3d_4')
group5 = f.get('conv3d_5')
g1 = group1.get('conv3d_1')
g2 = group2.get('conv3d_2')
g3 = group3.get('conv3d_3')
g4 = group4.get('conv3d_4')
g5 = group5.get('conv3d_5')
w1 = g1.get('kernel:0')
w1 = np.array(w1)
b1 = g1.get('bias:0')
b1 = np.array(b1)
w2 = g2.get('kernel:0')
w2 = np.array(w2)
b2 = g2.get('bias:0')
b2 = np.array(b2)
w3 = g3.get('kernel:0')
w3 = np.ar
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值