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

6 years ago · 554d3a71d2
parent 3acb3635c2
commit 554d3a71d2
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/fluid/tests/unittests/test_multiclass_nms_op.py
+++ b/python/paddle/fluid/tests/unittests/test_multiclass_nms_op.py
@ -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]