h5py读写

import os
import numpy as np
import cv2
import h5py


def save_image_to_h5py(path):
    f = h5py.File('data/h5/cat_or_not.h5', 'w')
    for idx, child_dir in enumerate(os.listdir(path)):
        # img = cv2.imread(os.path.join(path, child_dir)), (256, 256)
        img = cv2.resize(cv2.imread(os.path.join(path, child_dir)), (256, 256))
        f.create_dataset(str(idx), data=img)
    f.close()


save_image_to_h5py('/dataset/RAIN/real/')
with h5py.File('data/h5/cat_or_not.h5', "r") as f:
    for key in f.keys():
        print(f[key], '|*****|', key, '|*****|', f[key].name)
        cv2.imshow('img', np.array(f[key]))
        cv2.waitKey()
    f.close()