From 90e2362c003345d253fc699423056854fd1fa332 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Sat, 26 Dec 2020 13:50:27 -0700
Subject: [PATCH] Fix bug with full_image_dataset

---
 codes/data/full_image_dataset.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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