caffe使用python创建lmdb

import numpy as np
import sys
from PIL import Image
import lmdb
import random
import os

sys.path.append('../caffe/python/')

import caffe

if __name__ == '__main__' :
train_list_file = './trainlist.txt'
train_images_root = './train/images/'

f = open(train_list_file, 'r')
trainlist = f.readlines()
f.close()

random.shuffle(trainlist)

# creating images lmdb
in_db = lmdb.open('my-train-data-lmdb', map_size=int(1e12))
with in_db.begin(write=True) as in_txn :
for in_idx, in_ in enumerate(trainlist) :
fid = in_.strip()
fn = os.path.join(train_images_root, fid)
im = np.array(Image.open(fn))
im = im[:,:,::-1]
im = im.transpose((2, 0, 1))
im_dat = caffe.io.array_to_datum(im)
in_txn.put('{:0>10d}'.format(in_idx), im_dat.SerializeToString())
in_db.close()