未验证 提交 7721ef51 编写于 作者: W wawatt 提交者: GitHub

修复RandomErasingImage使用时参数报错 (#6180)

上级 999d6469
...@@ -322,7 +322,7 @@ class RandomErasingImage(BaseOperator): ...@@ -322,7 +322,7 @@ class RandomErasingImage(BaseOperator):
self.higher = higher self.higher = higher
self.aspect_ratio = aspect_ratio self.aspect_ratio = aspect_ratio
def apply(self, sample): def apply(self, sample, context=None):
gt_bbox = sample['gt_bbox'] gt_bbox = sample['gt_bbox']
im = sample['image'] im = sample['image']
if not isinstance(im, np.ndarray): if not isinstance(im, np.ndarray):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册