From aa16d88a8bf3608ba147cf79aed4cbfd7f098925 Mon Sep 17 00:00:00 2001
From: wangguanzhong <jerrywgz@126.com>
Date: Thu, 26 Nov 2020 16:02:52 +0800
Subject: [PATCH] fix mask_rcnn architecture (#1765)

---
 ppdet/modeling/architecture/mask_rcnn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppdet/modeling/architecture/mask_rcnn.py b/ppdet/modeling/architecture/mask_rcnn.py
index 9c542be3f..76b17998d 100644
--- a/ppdet/modeling/architecture/mask_rcnn.py
+++ b/ppdet/modeling/architecture/mask_rcnn.py
@@ -133,7 +133,7 @@ class MaskRCNN(BaseArch):
         loss.update({'loss': total_loss})
         return loss
 
-    def get_pred(self, ):
+    def get_pred(self, return_numpy=True):
         mask = self.mask_post_process(self.bboxes, self.mask_head_out,
                                       self.inputs['im_shape'],
                                       self.inputs['scale_factor'])
-- 
GitLab