From 7721ef510b6b7ae48b634d2e8b0c26af5218a9db Mon Sep 17 00:00:00 2001
From: wawatt <38525850+wawatt@users.noreply.github.com>
Date: Mon, 13 Jun 2022 15:31:25 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8DRandomErasingImage=E4=BD=BF?=
 =?UTF-8?q?=E7=94=A8=E6=97=B6=E5=8F=82=E6=95=B0=E6=8A=A5=E9=94=99=20(#6180?=
 =?UTF-8?q?)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 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 721f6d033..4ccd292ef 100644
--- 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):
-- 
GitLab