support BatchRandomResize for 2-D input (#3132)

d6f36b14 · wangguanzhong · GitHub · 7a732663 · d6f36b14
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

ppdet/data/transform/batch_operators.py ppdet/data/transform/batch_operators.py +2 -1

未找到文件。
--- 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