Non-Maximum Suppression

[i4yyds]

Hello, is there any non-maximal suppression module in the code?

[Madi200]

Non Maximal Suppression is used at Inference time. It's code is in test.py
https://github.com/hizhangp/yolo_tensorflow/blob/master/test.py
for i in range(len(boxes_filtered)): if probs_filtered[i] == 0: continue for j in range(i + 1, len(boxes_filtered)): if self.iou(boxes_filtered[i], boxes_filtered[j]) > self.iou_threshold: probs_filtered[j] = 0.0

    `filter_iou = np.array(probs_filtered > 0.0, dtype='bool')
    boxes_filtered = boxes_filtered[filter_iou]
    probs_filtered = probs_filtered[filter_iou]
    classes_num_filtered = classes_num_filtered[filter_iou]`