Python用HDF5存储和加载图片特征

        可以将大量图片数据存储于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存储与加载图片特征的方法,如有错误,欢迎指正。

你可能感兴趣的:(Python,python,hdf5,h5py)