From 63a8c6398d68ca884e7a4cce86d1123476a918b6 Mon Sep 17 00:00:00 2001 From: JieXiao Date: Wed, 21 Sep 2022 09:33:58 +0800 Subject: [PATCH] fix error --- dataset/imgdata.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/dataset/imgdata.py b/dataset/imgdata.py index 3017fd3..efca4c8 100644 --- a/dataset/imgdata.py +++ b/dataset/imgdata.py @@ -165,6 +165,9 @@ def __getitem__(self, index): tar_img = tar_img[:, rr:rr + self.crop_size, cc:cc + self.crop_size] # Data Augmentations aug = random.randint(0, 3) + # Crop patch + inp_img = inp_img[:, rr:rr + self.crop_size, cc:cc + self.crop_size] + tar_img = tar_img[:, rr:rr + self.crop_size, cc:cc + self.crop_size] if aug == 1: inp_img = inp_img.flip(1) tar_img = tar_img.flip(1)