Skip to content

P
PaddleDetection

s920243400 / PaddleDetection
与 Fork 源项目一致

Fork自 PaddlePaddle / PaddleDetection

通知 2
Star 0
Fork 0
P
PaddleDetection
未验证 提交 be0ca534 编写于 作者: W wangguanzhong 提交者： GitHub
浏览文件
操作

support BatchRandomResize for 2-D input (#3135)

上级 653ebe9f
隐藏空白更改
内联 并排
Showing with 2 addition and 1 deletion
ppdet/data/transform/batch_operators.py
浏览文件 @ be0ca534
...@@ -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.
先完成此消息的编辑！
想要评论请 注册 或