| 29 | |
| 30 | class LoadDatasetLmdb(Dataset): |
| 31 | def __init__(self,config,lmdb_file): |
| 32 | num_workers = config['train']['num_workers'] |
| 33 | self.fixKey = config['train']['fixKeyON'] |
| 34 | self.fixKeyType = config['train']['fixKeytype'] |
| 35 | assert self.fixKeyType in ['En','Ch'] |
| 36 | self.env = lmdb.open(lmdb_file, max_readers=num_workers, readonly=True, lock=False, readahead=False, meminit=False) |
| 37 | if not self.env: |
| 38 | print('cannot creat lmdb from %s' % (lmdb_file)) |
| 39 | sys.exit(0) |
| 40 | |
| 41 | with self.env.begin(write=False) as txn: |
| 42 | nSamples = int(txn.get('num-samples'.encode('utf-8'))) |
| 43 | self.nSamples = nSamples |
| 44 | |
| 45 | |
| 46 | def __len__(self): |