From b852c1c4334b191f3b816bac93406aa8a0188764 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=B0=A2=E6=98=95=E8=BE=B0?= Date: Wed, 23 Nov 2022 18:54:11 +0800 Subject: [PATCH] [Fix] Fix a not proper assertion in RandomChoiceResize (#2450) * fix assertion * remove assert * use is_seq_of --- mmcv/transforms/processing.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mmcv/transforms/processing.py b/mmcv/transforms/processing.py index fef4e608c..09098f199 100644 --- a/mmcv/transforms/processing.py +++ b/mmcv/transforms/processing.py @@ -1063,7 +1063,7 @@ class RandomChoiceResize(BaseTransform): self.scales = scales else: self.scales = [scales] - assert mmengine.is_list_of(self.scales, tuple) + assert mmengine.is_seq_of(self.scales, (tuple, int)) self.resize_cfg = dict(type=resize_type, **resize_kwargs) # create a empty Resize object @@ -1079,7 +1079,6 @@ class RandomChoiceResize(BaseTransform): ``scale_idx`` is the selected index in the given candidates. """ - assert mmengine.is_list_of(self.scales, tuple) scale_idx = np.random.randint(len(self.scales)) scale = self.scales[scale_idx] return scale, scale_idx