Skip to content

Commit

Permalink
pylint
Browse files Browse the repository at this point in the history
  • Loading branch information
yongwww committed Jun 18, 2020
1 parent 601c736 commit 5b8052d
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion topi/python/topi/vision/nms.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 \
Expand Down

0 comments on commit 5b8052d

Please sign in to comment.