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

7721ef51 · wawatt · GitHub · 999d6469 · 7721ef51
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

ppdet/data/transform/operators.py ppdet/data/transform/operators.py +1 -1

未找到文件。
--- a/ppdet/data/transform/operators.py
+++ b/ppdet/data/transform/operators.py
@@ -322,7 +322,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):