search

hizhangp / yolo_tensorflow

Tensorflow implementation of YOLO, including training and test phase.
MIT License
795 stars 442 forks source link

Non-Maximum Suppression #85

Open i4yyds opened 5 years ago

i4yyds commented 5 years ago

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

Madi200 commented 5 years ago

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]`