From f8dd14fdb0fa9f592ff32afcca4fe3c3e1492a48 Mon Sep 17 00:00:00 2001 From: Yong Wu Date: Fri, 19 Jun 2020 00:25:20 +0800 Subject: [PATCH] pylint --- topi/python/topi/vision/nms.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/topi/python/topi/vision/nms.py b/topi/python/topi/vision/nms.py index 2f2cc52207d5..1ee9e836705c 100644 --- a/topi/python/topi/vision/nms.py +++ b/topi/python/topi/vision/nms.py @@ -351,8 +351,9 @@ 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 - for k in parallel(j): + # check if current box j is valid by calculating iou with + # all existing valid boxes + for k in range(j): check_iou = 0 if is_valid_box == 1 and k < j and output[i, k, score_index] > 0 \ and (id_index < 0 or output[i, k, id_index] >= 0):