replace use_hflip with use_flip

Author: xinntao

basicsr/data/paired_image_dataset.py

@@ -32,7 +32,7 @@ class PairedImageDataset(data.Dataset):
             filename_tmpl (str): Template for each filename. Note that the
                 template excludes the file extension. Default: '{}'.
             gt_size (int): Cropped patched size for gt patches.
-            use_flip (bool): Use horizontal flips.
+            use_hflip (bool): Use horizontal flips.
             use_rot (bool): Use rotation (use vertical flip and transposing h
                 and w for implementation).
 
@@ -86,7 +86,7 @@ def __getitem__(self, index):
             # random crop
             img_gt, img_lq = paired_random_crop(img_gt, img_lq, gt_size, scale, gt_path)
             # flip, rotation
-            img_gt, img_lq = augment([img_gt, img_lq], self.opt['use_flip'], self.opt['use_rot'])
+            img_gt, img_lq = augment([img_gt, img_lq], self.opt['use_hflip'], self.opt['use_rot'])
 
         if 'color' in self.opt and self.opt['color'] == 'y':
             img_gt = rgb2ycbcr(img_gt, y_only=True)[..., None]

basicsr/data/reds_dataset.py

@@ -43,7 +43,7 @@ class REDSDataset(data.Dataset):
             gt_size (int): Cropped patched size for gt patches.
             interval_list (list): Interval list for temporal augmentation.
             random_reverse (bool): Random reverse input frames.
-            use_flip (bool): Use horizontal flips.
+            use_hflip (bool): Use horizontal flips.
             use_rot (bool): Use rotation (use vertical flip and transposing h
                 and w for implementation).
 
@@ -182,9 +182,9 @@ def __getitem__(self, index):
         # augmentation - flip, rotate
         img_lqs.append(img_gt)
         if self.flow_root is not None:
-            img_results, img_flows = augment(img_lqs, self.opt['use_flip'], self.opt['use_rot'], img_flows)
+            img_results, img_flows = augment(img_lqs, self.opt['use_hflip'], self.opt['use_rot'], img_flows)
         else:
-            img_results = augment(img_lqs, self.opt['use_flip'], self.opt['use_rot'])
+            img_results = augment(img_lqs, self.opt['use_hflip'], self.opt['use_rot'])
 
         img_results = img2tensor(img_results)
         img_lqs = torch.stack(img_results[0:-1], dim=0)
@@ -242,7 +242,7 @@ class REDSRecurrentDataset(data.Dataset):
             gt_size (int): Cropped patched size for gt patches.
             interval_list (list): Interval list for temporal augmentation.
             random_reverse (bool): Random reverse input frames.
-            use_flip (bool): Use horizontal flips.
+            use_hflip (bool): Use horizontal flips.
             use_rot (bool): Use rotation (use vertical flip and transposing h
                 and w for implementation).
 
@@ -345,7 +345,7 @@ def __getitem__(self, index):
 
         # augmentation - flip, rotate
         img_lqs.extend(img_gts)
-        img_results = augment(img_lqs, self.opt['use_flip'], self.opt['use_rot'])
+        img_results = augment(img_lqs, self.opt['use_hflip'], self.opt['use_rot'])
 
         img_results = img2tensor(img_results)
         img_gts = torch.stack(img_results[len(img_lqs) // 2:], dim=0)

basicsr/data/vimeo90k_dataset.py

@@ -42,7 +42,7 @@ class Vimeo90KDataset(data.Dataset):
             num_frame (int): Window size for input frames.
             gt_size (int): Cropped patched size for gt patches.
             random_reverse (bool): Random reverse input frames.
-            use_flip (bool): Use horizontal flips.
+            use_hflip (bool): Use horizontal flips.
             use_rot (bool): Use rotation (use vertical flip and transposing h
                 and w for implementation).
 
@@ -111,7 +111,7 @@ def __getitem__(self, index):
 
         # augmentation - flip, rotate
         img_lqs.append(img_gt)
-        img_results = augment(img_lqs, self.opt['use_flip'], self.opt['use_rot'])
+        img_results = augment(img_lqs, self.opt['use_hflip'], self.opt['use_rot'])
 
         img_results = img2tensor(img_results)
         img_lqs = torch.stack(img_results[0:-1], dim=0)
@@ -173,7 +173,7 @@ def __getitem__(self, index):
 
         # augmentation - flip, rotate
         img_lqs.extend(img_gts)
-        img_results = augment(img_lqs, self.opt['use_flip'], self.opt['use_rot'])
+        img_results = augment(img_lqs, self.opt['use_hflip'], self.opt['use_rot'])
 
         img_results = img2tensor(img_results)
         img_lqs = torch.stack(img_results[:7], dim=0)

docs/Config.md

@@ -71,7 +71,7 @@ datasets:
     # Ground-Truth training patch size
     gt_size: 128
     # Whether to use horizontal flip. Here, flip is for horizontal flip
-    use_flip: true
+    use_hflip: true
     # Whether to rotate. Here for rotations with every 90 degree
     use_rot: true
 

docs/Config_CN.md

@@ -71,7 +71,7 @@ datasets:
     # 训练中Ground-Truth的Training patch的大小
     gt_size: 128
     # 是否使用horizontal flip, 这里的flip特指 horizontal flip
-    use_flip: true
+    use_hflip: true
     # 是否使用rotation, 这里指的是每隔90°旋转
     use_rot: true
 

options/train/BasicVSR/train_BasicVSR_REDS.yml

@@ -22,7 +22,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/BasicVSR/train_BasicVSR_Vimeo90K_BDx4.yml

@@ -20,7 +20,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
     flip_sequence: true
 

options/train/BasicVSR/train_BasicVSR_Vimeo90K_BIx4.yml

@@ -20,7 +20,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
     flip_sequence: true
 

options/train/BasicVSR/train_IconVSR_REDS.yml

@@ -22,7 +22,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/BasicVSR/train_IconVSR_Vimeo90K_BDx4.yml

@@ -20,7 +20,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
     flip_sequence: true
 

options/train/BasicVSR/train_IconVSR_Vimeo90K_BIx4.yml

@@ -20,7 +20,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
     flip_sequence: true
 

options/train/ECBSR/train_ECBSR_x2_m4c16_prelu.yml

@@ -19,7 +19,7 @@ datasets:
       type: disk
 
     gt_size: 128
-    use_flip: true
+    use_hflip: true
     use_rot: true
     color: y
 

options/train/ECBSR/train_ECBSR_x4_m4c16_prelu.yml

@@ -19,7 +19,7 @@ datasets:
       type: lmdb
 
     gt_size: 256
-    use_flip: true
+    use_hflip: true
     use_rot: true
     color: y
 

options/train/ECBSR/train_ECBSR_x4_m4c16_prelu_RGB.yml

@@ -19,7 +19,7 @@ datasets:
       type: lmdb
 
     gt_size: 256
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/EDSR/train_EDSR_Lx2.yml

@@ -22,7 +22,7 @@ datasets:
       # type: lmdb
 
     gt_size: 96
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/EDSR/train_EDSR_Lx3.yml

@@ -22,7 +22,7 @@ datasets:
       # type: lmdb
 
     gt_size: 144
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/EDSR/train_EDSR_Lx4.yml

@@ -22,7 +22,7 @@ datasets:
       # type: lmdb
 
     gt_size: 192
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/EDSR/train_EDSR_Mx2.yml

@@ -22,7 +22,7 @@ datasets:
       # type: lmdb
 
     gt_size: 96
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/EDSR/train_EDSR_Mx3.yml

@@ -22,7 +22,7 @@ datasets:
       # type: lmdb
 
     gt_size: 144
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/EDSR/train_EDSR_Mx4.yml

@@ -22,7 +22,7 @@ datasets:
       # type: lmdb
 
     gt_size: 192
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/EDVR/train_EDVRM_woTSA_GAN_TODO.yml

@@ -23,7 +23,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/EDVR/train_EDVR_L_x4_SR_REDS.yml

@@ -22,7 +22,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/EDVR/train_EDVR_L_x4_SR_REDS_woTSA.yml

@@ -22,7 +22,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/EDVR/train_EDVR_M_x4_SR_REDS.yml

@@ -22,7 +22,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/EDVR/train_EDVR_M_x4_SR_REDS_woTSA.yml

@@ -22,7 +22,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/ESRGAN/train_ESRGAN_x4.yml

@@ -22,7 +22,7 @@ datasets:
       # type: lmdb
 
     gt_size: 128
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/ESRGAN/train_RRDBNet_PSNR_x4.yml

@@ -22,7 +22,7 @@ datasets:
       # type: lmdb
 
     gt_size: 128
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/HiFaceGAN/train_hifacegan.yml

@@ -20,7 +20,7 @@ datasets:
       # type: lmdb
 
     gt_size: 512
-    use_flip: true
+    use_hflip: true
     use_rot: false
 
     # data loader

options/train/RCAN/train_RCAN_x2.yml

@@ -23,7 +23,7 @@ datasets:
       # type: lmdb
 
     gt_size: 96
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/SRResNet_SRGAN/train_MSRGAN_x4.yml

@@ -26,7 +26,7 @@ datasets:
       # type: lmdb
 
     gt_size: 128
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/SRResNet_SRGAN/train_MSRResNet_x2.yml

@@ -25,7 +25,7 @@ datasets:
       # type: lmdb
 
     gt_size: 64
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/SRResNet_SRGAN/train_MSRResNet_x3.yml

@@ -25,7 +25,7 @@ datasets:
       # type: lmdb
 
     gt_size: 96
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/SRResNet_SRGAN/train_MSRResNet_x4.yml

@@ -26,7 +26,7 @@ datasets:
       # type: lmdb
 
     gt_size: 128
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/SwinIR/train_SwinIR_SRx2_scratch.yml

@@ -18,7 +18,7 @@ datasets:
       type: disk
 
     gt_size: 96
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/SwinIR/train_SwinIR_SRx4_scratch.yml

@@ -18,7 +18,7 @@ datasets:
       type: disk
 
     gt_size: 192
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

options/train/VideoRecurrentGAN/train_VideoRecurrentGANModel_REDS.yml

@@ -23,7 +23,7 @@ datasets:
     gt_size: 256
     interval_list: [1]
     random_reverse: false
-    use_flip: true
+    use_hflip: true
     use_rot: true
 
     # data loader

test_scripts/test_paired_image_dataset.py

@@ -34,7 +34,7 @@ def main(mode='folder'):
         opt['io_backend'] = dict(type='lmdb')
 
     opt['gt_size'] = 128
-    opt['use_flip'] = True
+    opt['use_hflip'] = True
     opt['use_rot'] = True
 
     opt['use_shuffle'] = True

test_scripts/test_reds_dataset.py

@@ -35,7 +35,7 @@ def main(mode='folder'):
     opt['gt_size'] = 256
     opt['interval_list'] = [1]
     opt['random_reverse'] = True
-    opt['use_flip'] = True
+    opt['use_hflip'] = True
     opt['use_rot'] = True
 
     opt['use_shuffle'] = True