prepare_data.py 文件源码

def read_data(dataFile,size,Transpose=False,resize=False):
    f             =      h5py.File(dataFile,'r')
    result_data   =      np.zeros(size)
    result_label  =      np.zeros(size)
    if Transpose == True:
        data      =      np.transpose(np.array(f['data']),(3,2,1,0))
        label     =      np.transpose(np.array(f['label']),(3,2,1,0))
    else:
        data      =      np.array(f['data'])
        label     =      np.array(f['label'])
    [d1,d2,d3,d4] =      data.shape
    if resize == True:
        for p in range(d1):
            for d in range(d2):
                result_data[p,d,:,:]     =      np.resize(data[p,d,:,:],(size[2],size[3]))
                result_label[p,d,:,:]    =      np.resize(label[p,d,:,:],(size[2],size[3]))
        data      =      result_data
        label     =      result_label
    return data, label