未验证 提交 be0ca534 编写于 作者: W wangguanzhong 提交者: GitHub

support BatchRandomResize for 2-D input (#3135)

上级 653ebe9f
...@@ -136,7 +136,8 @@ class BatchRandomResize(BaseOperator): ...@@ -136,7 +136,8 @@ class BatchRandomResize(BaseOperator):
def __call__(self, samples, context=None): def __call__(self, samples, context=None):
if self.random_size: 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: else:
target_size = self.target_size target_size = self.target_size
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册