a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Allow multiple gt image dirs

Browse Source
This commit is contained in:
James Betker 2020-05-25 19:21:09 -06:00
parent 4e44b8a1aa
commit 2931142458
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
codes/data/util.py
View File

@ -50,7 +50,13 @@ def get_image_paths(data_type, dataroot):
if data_type == 'lmdb': if data_type == 'lmdb':
paths, sizes = _get_paths_from_lmdb(dataroot) paths, sizes = _get_paths_from_lmdb(dataroot)
elif data_type == 'img': elif data_type == 'img':
paths = sorted(_get_paths_from_images(dataroot)) if isinstance(dataroot, list):
paths = []
for r in dataroot:
paths.extend(_get_paths_from_images(r))
paths = sorted(paths)
else:
paths = sorted(_get_paths_from_images(dataroot))
else: else:
raise NotImplementedError('data_type [{:s}] is not recognized.'.format(data_type)) raise NotImplementedError('data_type [{:s}] is not recognized.'.format(data_type))
return paths, sizes return paths, sizes