Fix

open-mmlab · Jun 15, 2021 · 0d1f196 · 0d1f196
1 parent 6404db2
commit 0d1f196
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/mmedit/models/restorers/dic.py b/mmedit/models/restorers/dic.py
@@ -53,8 +53,7 @@ def __init__(self,
         self.align_loss = build_loss(align_loss)
 
         # pretrained
-        if pretrained:
-            self.init_weights(pretrained)
+        self.init_weights(pretrained)
 
     def forward(self, lq, gt=None, test_mode=False, **kwargs):
         """Forward function.
@@ -167,10 +166,10 @@ def forward_test(self,
         # save image
         if save_image:
             if 'gt_path' in meta[0]:
-                the_path = meta[0]['gt_path']
+                pred_path = meta[0]['gt_path']
             else:
-                the_path = meta[0]['lq_path']
-            folder_name = osp.splitext(osp.basename(the_path))[0]
+                pred_path = meta[0]['lq_path']
+            folder_name = osp.splitext(osp.basename(pred_path))[0]
             if isinstance(iteration, numbers.Number):
                 save_path = osp.join(save_path, folder_name,
                                      f'{folder_name}-{iteration + 1:06d}.png')