未验证 提交 b8f70f95 编写于 作者: C channings 提交者: GitHub

fix bug of exclud_nms (#2275)

上级 01d905e3
......@@ -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])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册