From ae54352f57f397df6f9da5c2c87ee54aa486a9be Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Wed, 6 Apr 2022 14:31:55 +0800 Subject: [PATCH] fix rcnn pred (#5598) --- ppdet/modeling/architectures/faster_rcnn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppdet/modeling/architectures/faster_rcnn.py b/ppdet/modeling/architectures/faster_rcnn.py index 26a2672d6..ce9a8e4b5 100644 --- a/ppdet/modeling/architectures/faster_rcnn.py +++ b/ppdet/modeling/architectures/faster_rcnn.py @@ -87,8 +87,8 @@ class FasterRCNN(BaseArch): im_shape, scale_factor) # rescale the prediction back to origin image - bbox_pred = self.bbox_post_process.get_pred(bbox, bbox_num, - im_shape, scale_factor) + bboxes, bbox_pred, bbox_num = self.bbox_post_process.get_pred( + bbox, bbox_num, im_shape, scale_factor) return bbox_pred, bbox_num def get_loss(self, ): -- GitLab