可以将大量图片数据存储于hdf5文件中,每张图片以dataset的形式存储。
import h5py
import numpy as np
image_features = np.random.randn(512) # 生成的图像特征
with h5py.File('img.hdf5', 'w') as f:
# 每一条图像特征作为dataset存储
dset = f.create_dataset("img_name", data=image_features)
读取的时候根据dataset的name去检索出对应的数据,存储和加载出来的格式相同。
with h5py.File('img.hdf5', 'r') as h:
# 根据dataset的name检索出对应的数据
image_features = h["img_name"][()]
以上是简易的HDF5存储与加载图片特征的方法,如有错误,欢迎指正。