From 5b8052d8866d64af663c9908d094f48f52e845c5 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 | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 \