未验证 提交 ec3b094e 编写于 作者: F Feng Ni 提交者: GitHub

fix RandomErasingImage apply (#7441)

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