diff --git a/codes/data/full_image_dataset.py b/codes/data/full_image_dataset.py index 9e121f27..fff6cf3f 100644 --- a/codes/data/full_image_dataset.py +++ b/codes/data/full_image_dataset.py @@ -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 \ No newline at end of file + pass