lmdb数据库的使用

import lmdb

db_path = 'lmdb_dir'
env = lmdb.open(db_path, subdir=osp.isdir(db_path),
                                 readonly=True, lock=False,
                                 readahead=False, meminit=False)
env.close()


自己创建好数据库之后就可以使用了:
ITS_train_loader_lmdb = DataLoader(
    dataset=DatasetLMDB(os.path.join(path, 'ITS/DATA.mdb'), size=crop_size), batch_size=BS,
    shuffle=True, pin_memory=True)

你可能感兴趣的:(数据库)