From b8f70f955fdc5d9ee6c617349770282fb30bbc50 Mon Sep 17 00:00:00 2001
From: channings <chenlingchi@baidu.com>
Date: Tue, 9 Mar 2021 16:53:16 +0800
Subject: [PATCH] fix bug of exclud_nms (#2275)

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

diff --git a/ppdet/modeling/anchor_heads/yolo_head.py b/ppdet/modeling/anchor_heads/yolo_head.py
index df8a375cf..645b1e835 100644
--- a/ppdet/modeling/anchor_heads/yolo_head.py
+++ b/ppdet/modeling/anchor_heads/yolo_head.py
@@ -458,7 +458,7 @@ class YOLOv3Head(object):
 
         # Only for benchmark, postprocess(NMS) is not needed
         if exclude_nms:
-            return {'bbox': yolo_scores}
+            return {'bbox': yolo_boxes, 'score': yolo_scores}
 
         if type(self.nms) is MultiClassSoftNMS:
             yolo_scores = fluid.layers.transpose(yolo_scores, perm=[0, 2, 1])
-- 
GitLab