mrbi_to_lmdb.py 文件源码

def view_lmdb_data():
    lmdb_env = lmdb.open('/home/lisha/school/Projects/hyperband_nnet/hyperband2/svhn/svhn_train/')
    lmdb_txn = lmdb_env.begin()
    lmdb_cursor = lmdb_txn.cursor()
    datum = caffe.proto.caffe_pb2.Datum()
    x=[]
    y=[]

    for key, value in lmdb_cursor:
        datum.ParseFromString(value)
        label = datum.label
        data = caffe.io.datum_to_array(datum)
        #plt.imshow(np.rollaxis(data,0,3))
        x.append(data)
        y.append(label)
    print len(y)