提交 554d3a71 编写于 作者: J Jiancheng Li 提交者: lujun

test=develop fix bug: fix selected_indices in nms (#17140)

上级 3acb3635
......@@ -81,6 +81,7 @@ def nms(boxes,
sorted_indices = np.argsort(-all_scores, axis=0, kind='mergesort')
sorted_scores = all_scores[sorted_indices]
sorted_indices = selected_indices[sorted_indices]
if top_k > -1 and top_k < sorted_indices.shape[0]:
sorted_indices = sorted_indices[:top_k]
sorted_scores = sorted_scores[:top_k]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册