diff --git a/ppdet/data/transform/batch_operators.py b/ppdet/data/transform/batch_operators.py index 253eae815d045d2e70ea05ae1e751dedef42581a..fff73d52b336c4f92c59093f62d843d576e167cc 100644 --- a/ppdet/data/transform/batch_operators.py +++ b/ppdet/data/transform/batch_operators.py @@ -136,7 +136,8 @@ class BatchRandomResize(BaseOperator): def __call__(self, samples, context=None): if self.random_size: - target_size = np.random.choice(self.target_size) + index = np.random.choice(len(self.target_size)) + target_size = self.target_size[index] else: target_size = self.target_size