未验证 提交 76ab2780 编写于 作者: W wangguanzhong 提交者: GitHub

fix return_index in nms (#2980)

上级 33c8aebd
...@@ -987,7 +987,7 @@ def multiclass_nms(bboxes, ...@@ -987,7 +987,7 @@ def multiclass_nms(bboxes,
'normalized', normalized) 'normalized', normalized)
output, index, nms_rois_num = core.ops.multiclass_nms3(bboxes, scores, output, index, nms_rois_num = core.ops.multiclass_nms3(bboxes, scores,
rois_num, *attrs) rois_num, *attrs)
if return_index: if not return_index:
index = None index = None
return output, nms_rois_num, index return output, nms_rois_num, index
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册