Fix bug with full_image_dataset

This commit is contained in:
James Betker 2020-12-26 13:50:27 -07:00
parent 10fdfa1563
commit 90e2362c00

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,
'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
def __len__(self):
@ -371,4 +371,4 @@ if __name__ == '__main__':
#v = v[:3, :, :]
#import torchvision
#torchvision.utils.save_image(v.unsqueeze(0), "debug/%i_%s.png" % (i, k))
pass
pass