Commit 5d328c5c by mbarrett97 Committed by Yao Wang

[TOPI] Fixed nms max_output_size loop (#4541)

One of the loops in hybrid_nms used for
performing the max_output_size reordering
was incorrectly designated as parallel
resulting in incorrect behaviour. This patch
changes that loop to a serial loop.

Change-Id: I97184f5887f5f028d8ab339fa2808eb7630a4017
parent 8ba9e8bc
......@@ -278,7 +278,7 @@ def hybrid_nms(data, sorted_index, valid_count,
# Only return max_output_size valid boxes
num_valid_boxes = 0
if max_output_size > 0:
for j in parallel(valid_count[i]):
for j in range(valid_count[i]):
if output[i, j, 0] >= zero:
if num_valid_boxes == max_output_size:
for k in range(box_data_length):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment