diff --git a/topi/python/topi/vision/nms.py b/topi/python/topi/vision/nms.py index 2f2cc52207d54..89d9a40f43d9a 100644 --- a/topi/python/topi/vision/nms.py +++ b/topi/python/topi/vision/nms.py @@ -351,7 +351,8 @@ def hybrid_nms(data, sorted_index, valid_count, indices, batch_size, num_anchors a_b = max(output[batch_idx, box_a_idx, box_start_idx + 1], output[batch_idx, box_a_idx, box_start_idx + 3]) - # check if current box j is valid by calculating iou with all existing valid boxes + # check if current box j is valid by calculating iou with + # all existing valid boxes for k in parallel(j): check_iou = 0 if is_valid_box == 1 and k < j and output[i, k, score_index] > 0 \