mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Disable LMDB support

Browse Source 
It doesn't play nice with multiple dataroots and I don't
really see any need to continue support since I'm not
testing it.
This commit is contained in:
James Betker 2020-05-13 15:27:33 -06:00
parent e36f22e14a
commit 5bcf187fb6
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/options/options.py
View File

@ -26,6 +26,7 @@ def parse(opt_path, is_train=True):
if opt['distortion'] == 'sr' or opt['distortion'] == 'downsample':
dataset['scale'] = scale
is_lmdb = False
''' LMDB is not supported at this point with the mods I've been making.
if dataset.get('dataroot_GT', None) is not None:
dataset['dataroot_GT'] = osp.expanduser(dataset['dataroot_GT'])
if dataset['dataroot_GT'].endswith('lmdb'):
@ -34,6 +35,7 @@ def parse(opt_path, is_train=True):
dataset['dataroot_LQ'] = osp.expanduser(dataset['dataroot_LQ'])
if dataset['dataroot_LQ'].endswith('lmdb'):
is_lmdb = True
'''
dataset['data_type'] = 'lmdb' if is_lmdb else 'img'
if dataset['mode'].endswith('mc'): # for memcached
dataset['data_type'] = 'mc'