mrbi_to_lmdb.py 文件源码

def make_test():
    print 'Loading Matlab data.'
    f = '/home/lisha/school/Projects/hyperband_nnet/hyperband2/mrbi/mnist_rotation_back_image_new/mnist_all_background_images_rotation_normalized_test.amat'

    # name of your matlab variables:

    X,Y=get_data(f)
    N = Y.shape[0]
    map_size = X.nbytes*2
    #if you want to shuffle your data
    #random.shuffle(N)
    env = lmdb.open('/home/lisha/school/Projects/hyperband_nnet/hyperband2/mrbi/mrbi_test', map_size=map_size)
    with env.begin(write=True) as txn:
        # txn is a Transaction object
        for i in range(N):
            im_dat = caffe.io.array_to_datum(X[i],Y[i])
            txn.put('{:0>10d}'.format(i), im_dat.SerializeToString())