diff --git a/ppdet/modeling/anchor_heads/yolo_head.py b/ppdet/modeling/anchor_heads/yolo_head.py index df8a375cf18c94f7d0e0746ece9c4a3ab48f58b7..645b1e835bf1bc08515d36ff4aa079819b077506 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])