--wip-- [skip ci]

Author: ericHao22

mini_batch_loader.py

@@ -36,8 +36,8 @@ def load_training_data(self, index):
     def load_data(self, path_infos, index):
         in_channels = 3
         path = path_infos[index]
-        mask_path = path.replace('input', 'RDP_w_sampling')
-        pseudo_gt_dir = path.replace('input', 'Pseudo_Reference_RDP_w_sampling')[:-4]
+        mask_path = path.replace('input', 'RDP')
+        pseudo_gt_dir = path.replace('input', 'Pseudo_Reference_RDP')[:-4]
 
         img = cv2.imread(path)
         if '.jpg' in path:
@@ -78,8 +78,8 @@ def load_batch_data(self, path_infos, indices, img_size=None, augment=False):
 
             for i, index in enumerate(indices):
                 path = path_infos[index]
-                mask_path = path.replace('input', 'RDP_w_sampling')
-                pseudo_gt_dir = path.replace('input', 'Pseudo_Reference_RDP_w_sampling')[:-4]
+                mask_path = path.replace('input', 'RDP')
+                pseudo_gt_dir = path.replace('input', 'Pseudo_Reference_RDP')[:-4]
 
                 img = cv2.imread(path)
                 if '.jpg' in path:
@@ -128,7 +128,7 @@ def load_batch_data(self, path_infos, indices, img_size=None, augment=False):
         elif mini_batch_size == 1:
             for i, index in enumerate(indices):
                 path = path_infos[index]
-                mask_path = path.replace('input', 'RDP_w_sampling')
+                mask_path = path.replace('input', 'RDP')
 
                 img = cv2.imread(path)
                 if '.jpg' in path:

stochastic_filling.py

@@ -5,7 +5,7 @@
 
 dataset_path = './dataset/Rain100L/test/'
 save_path = './dataset/Rain100L/test/'
-target_path = 'Pseudo_Reference_RDP_w_sampling/'
+target_path = 'Pseudo_Reference_RDP/'
 
 def make_folder(path):
    try:
@@ -42,7 +42,7 @@ def compute_similarity(rainy_image, rdp_image, j, i):
 
 ### MAIN PROCESS GOES HERE ###
 rainy_path = os.path.join(dataset_path, "input")
-rdp_path = os.path.join(dataset_path, "RDP_w_sampling")
+rdp_path = os.path.join(dataset_path, "RDP")
 rainy_folder = os.listdir(rainy_path)
 print(rainy_folder)
 print(len(rainy_folder))