09b8ead9c0aa8d1fb635c2812dddacd8645ccf08,codes/scripts/create_lmdb.py,,,#,9

Before Change

dataset = []
data_size = 0
print("Read image...")
for i in range(len(file_list)):
    if i % 100 == 0:
        print(i)
    img = cv2.imread(file_list[i], cv2.IMREAD_UNCHANGED)
    dataset.append(img)
    data_size += img.nbytes
map_size = data_size * 10

After Change

print("Finish reading image - {}.\nWrite to lmdb...".format(len(file_list)))
with env.begin(write=True) as txn:  // txn is a Transaction object
    for i, v in enumerate(file_list):
        if i % 100 == 0:
            print(i)
        img_name = os.path.splitext(os.path.basename(v))[0]

Instances