caffe hdf5 数据层数据生成

使用的是python里面带的h5py这个库，
使用方式是通过hy.File(filename，mode)读取hdf5或者写入hdf5。
输入存入的方式是通过字典赋值的方式进行的。

import h5py as hy
import numpy as np
import cv2
width=100
height=224
fid=open('/home/yang/Desktop/all.txt')
lines=fid.readlines()
f = hy.File('HDF5_FILE.h5','w')
imgData=np.zeros((2,3,height,width))
labels=np.zeros((2,3))
idx=0
for line in lines:
    line=line.strip().split()
    image=cv2.imread(line[0])
    image=cv2.resize(image,(width,height))
    image=image[:,:,[2,1,0]]
    image=image.transpose((2,0,1))
    imgData[idx,:,:,:]=image
    labels[idx,0]=int(line[1])
    labels[idx,1]=int(line[2])
    labels[idx,2]=int(line[3])
    idx+=1
f['data1']=imgData #这个地方的名字决定了你使用caffe时候top的名称，名称没对应上是错误的。
f['label1']=labels #同上面
print labels
fid.close()
f.close()