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

LQGT_dataset gan debug

Browse Source
This commit is contained in:
James Betker 2020-07-26 22:48:35 -06:00
parent d09ed4e5f7
commit 0892d5fe99
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/data/LQGT_dataset.py
View File

@ -50,6 +50,7 @@ class LQGTDataset(data.Dataset):
# LR and HR do not need to be paired. # LR and HR do not need to be paired.
if 'dataroot_GAN' in opt.keys(): if 'dataroot_GAN' in opt.keys():
self.paths_GAN, self.sizes_GAN = util.get_image_paths(self.data_type, opt['dataroot_GAN']) self.paths_GAN, self.sizes_GAN = util.get_image_paths(self.data_type, opt['dataroot_GAN'])
print('loaded %i images for use in training GAN only.' % (self.sizes_GAN,))
assert self.paths_GT, 'Error: GT path is empty.' assert self.paths_GT, 'Error: GT path is empty.'
if self.paths_LQ and self.paths_GT: if self.paths_LQ and self.paths_GT: