fix cascade infer (#1877)

c5c91cee · wangguanzhong · GitHub · a1446709 · c5c91cee
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

ppdet/modeling/architecture/cascade_rcnn.py ppdet/modeling/architecture/cascade_rcnn.py +1 -1

未找到文件。
--- 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