fix: normalize redundant loop (open-mmlab#121)

Yshuo-Li · Jul 26, 2020 · d152e15 · d152e15
1 parent 7794b00
commit d152e15
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/mmedit/datasets/pipelines/normalization.py b/mmedit/datasets/pipelines/normalization.py
@@ -37,11 +37,10 @@ def __call__(self, results):
         """
         for key in self.keys:
             if isinstance(results[key], list):
-                for v in results[key]:
-                    results[key] = [
-                        mmcv.imnormalize(v, self.mean, self.std, self.to_rgb)
-                        for v in results[key]
-                    ]
+                results[key] = [
+                    mmcv.imnormalize(v, self.mean, self.std, self.to_rgb)
+                    for v in results[key]
+                ]
             else:
                 results[key] = mmcv.imnormalize(results[key], self.mean,
                                                 self.std, self.to_rgb)