From ec3b094efe0fb93d91dbcef45cd0f771f79b385b Mon Sep 17 00:00:00 2001
From: Feng Ni <nemonameless@qq.com>
Date: Mon, 5 Dec 2022 12:02:51 +0800
Subject: [PATCH] fix RandomErasingImage apply (#7441)

---
 ppdet/data/transform/operators.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py
index b706a43ba..8c6a65e7b 100644
--- a/ppdet/data/transform/operators.py
+++ b/ppdet/data/transform/operators.py
@@ -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):
-- 
GitLab