From c5c91ceec77ff9abc594e29d35988820c4425d0a Mon Sep 17 00:00:00 2001
From: wangguanzhong <jerrywgz@126.com>
Date: Fri, 11 Dec 2020 15:46:52 +0800
Subject: [PATCH] fix cascade infer (#1877)

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

diff --git a/ppdet/modeling/architecture/cascade_rcnn.py b/ppdet/modeling/architecture/cascade_rcnn.py
index 1a678b6a0..1cfdaea51 100644
--- a/ppdet/modeling/architecture/cascade_rcnn.py
+++ b/ppdet/modeling/architecture/cascade_rcnn.py
@@ -165,5 +165,5 @@ class CascadeRCNN(BaseArch):
             'bbox_num': bbox_num,
         }
         if self.with_mask:
-            output.update(self.mask_head_out)
+            output.update({'mask': self.mask_head_out})
         return output
-- 
GitLab