create HDF5 dataset

h5test.py

# coding: utf-8
import cv2
import h5py
import numpy

img = cv2.imread('/Users/chexov/opencv-lena.jpg', 1)

f = h5py.File('test2.hdf5', mode='w')

group = f.create_group("images")
dset = group.create_dataset("img1", shape=(512,512,3), dtype=numpy.uint8, data=img)
dset.attrs['CLASS'] = 'IMAGE'
dset.attrs['IMAGE_VERSION'] = '1.2'
dset.attrs['IMAGE_MINMAXRANGE'] = numpy.array([0,255], dtype=numpy.uint8)
dset.attrs['IMAGE_SUBCLASS'] =  'IMAGE_TRUECOLOR'
dset.attrs['INTERLACE_MODE'] =  'INTERLACE_PIXEL'

group = f.create_group("imagebatches")
dset = group.create_dataset("imgbatch", shape=(4, 512,512,3), dtype=numpy.uint8)
dset.attrs['CLASS'] = 'IMAGE'
dset.attrs['IMAGE_VERSION'] = '1.2'
dset.attrs['IMAGE_MINMAXRANGE'] = numpy.array([0,255], dtype=numpy.uint8)
dset.attrs['IMAGE_SUBCLASS'] =  'IMAGE_TRUECOLOR'
dset.attrs['INTERLACE_MODE'] =  'INTERLACE_PIXEL'
dset[0, ...] = img

f.close()