minor fix

open-mmlab · Sep 14, 2021 · c68f54c · c68f54c
1 parent 7195cd7
commit c68f54c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmedit/datasets/pipelines/random_degradations.py b/mmedit/datasets/pipelines/random_degradations.py
@@ -140,7 +140,7 @@ def _random_resize(self, imgs):
         target_size = self.params.get('target_size', None)
         if target_size is None:
             resize_mode = np.random.choice(['up', 'down', 'keep'],
-                                           self.params['resize_mode_prob'])
+                                           p=self.params['resize_mode_prob'])
             resize_scale = self.params['resize_scale']
             if resize_mode == 'up':
                 scale_factor = np.random.uniform(1, resize_scale[1])