fix rcnn pred (#5598)

ae54352f · wangguanzhong · GitHub · e23ab8d8 · ae54352f
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

ppdet/modeling/architectures/faster_rcnn.py ppdet/modeling/architectures/faster_rcnn.py +2 -2

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