m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

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

Browse Source
feature/fluid_trt_int8
Jiancheng Li 6 years ago committed by lujun
parent 3acb3635c2
commit 554d3a71d2
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File

1
python/paddle/fluid/tests/unittests/test_multiclass_nms_op.py
Unescape View File

@ -81,6 +81,7 @@ def nms(boxes,
sorted_indices = np.argsort(-all_scores, axis=0, kind='mergesort') sorted_indices = np.argsort(-all_scores, axis=0, kind='mergesort')
sorted_scores = all_scores[sorted_indices] sorted_scores = all_scores[sorted_indices]
sorted_indices = selected_indices[sorted_indices]
if top_k > -1 and top_k < sorted_indices.shape[0]: if top_k > -1 and top_k < sorted_indices.shape[0]:
sorted_indices = sorted_indices[:top_k] sorted_indices = sorted_indices[:top_k]
sorted_scores = sorted_scores[:top_k] sorted_scores = sorted_scores[:top_k]

Write Preview
Loading…
Cancel
Save