在用深度学习训练的时候，我们经常要保存一些数据，用pickle的话，最大只能保存４GB，但是hdf5是没有限制的，所以这里介绍一下如果用python的h5py库来读取数据和写入数据。
首先要安装h5py:
pip install h5py写入
import numpy as np
import h5py
data_to_write = np.random.random(size=(100,100))
with h5py.File('name-of-file.h5', 'w') as hf:
hf.create_dataset("name-of-dataset", data=data_to_write)读取
with h5py.File('name-of-file.h5', 'r') as hf:
data = hf['name-of-dataset'][:]参考文献
[1].Input and output numpy arrays to h5py. https://stackoverflow.com/questions/20928136/input-and-output-numpy-arrays-to-h5py
扫一扫
1599
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
