mirror of https://github.com/open-mmlab/mmcv.git
[Fix] Fix a not proper assertion in RandomChoiceResize (#2450)
* fix assertion * remove assert * use is_seq_ofpull/2374/merge
parent
a0939977a3
commit
b852c1c433
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue