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

Fix bug with full_image_dataset

Browse Source
This commit is contained in:
James Betker 2020-12-26 13:50:27 -07:00
parent 10fdfa1563
commit 90e2362c00
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/data/full_image_dataset.py
View File

@ -325,7 +325,7 @@ class FullImageDataset(data.Dataset):
d = {'lq': img_LQ, 'hq': img_GT, 'gt_fullsize_ref': gt_fullsize_ref, 'lq_fullsize_ref': lq_fullsize_ref, d = {'lq': img_LQ, 'hq': img_GT, 'gt_fullsize_ref': gt_fullsize_ref, 'lq_fullsize_ref': lq_fullsize_ref,
'lq_center': lq_center, 'gt_center': gt_center, 'lq_center': lq_center, 'gt_center': gt_center,
'LQ_path': LQ_path, 'GT_path': full_path} 'LQ_path': LQ_path, 'HQ_path': full_path}
return d return d
def __len__(self): def __len__(self):
@ -371,4 +371,4 @@ if __name__ == '__main__':
#v = v[:3, :, :] #v = v[:3, :, :]
#import torchvision #import torchvision
#torchvision.utils.save_image(v.unsqueeze(0), "debug/%i_%s.png" % (i, k)) #torchvision.utils.save_image(v.unsqueeze(0), "debug/%i_%s.png" % (i, k))
pass pass