笔记地址：有道云笔记

h5py 的安装

pip install h5py

读取 H5 文件

import h5py import numpy as np # 打开文件 f = h5py.File('test-dev.h5', 'r')

H5 中的group可以类比为字典，因此我们可以用keys()来获取键值。

>>> f.keys() [u'my_xmax', u'my_xmin', u'my_ymax', u'my_ymin']

如上，我们发现文件里有四个数据集，我们可以像读取字典一样读取文件。

>>> xmax = f['my_xmax']

读取到的xmax，可以像处理numpy矩阵那样操作。这里有个小窍门。

>> xmax = f['my_xmax'] >>> type(xmax) h5py._hl.dataset.Dataset >>> xmax = f['my_xmax'][:] >>> type(xmax) numpy.ndarray

写入H5文件

打开方式用'w'，利用文件方法写入数据

f = h5py.File('test-dev.h5','w') f.create_dataset('bndbox', data=h5_bndbox) f.create_dataset('imgname', data=h5_imgname) f.create_dataset('part', data=h5_part)

又或者是直接的赋值

f['bndbox'] = h5_bndbox f['imgname'] = h5_imgname f['part'] = h5_part

关闭文件

f.close()