yolo预测出了多个边界框,我们选择哪一个呢?
不可以选择最大概率的框。以狗狗举例,概率分别为0.9;0.7;0.6,若画面中还有其他的狗狗的概率框为0.91,那么就会漏掉所见到的可爱的金毛。会漏检。
采用IOU交并比的NMS(非极大值抑制)解决(Intersection over Union)
yolo会把最大概率的边界框先拉出来,和其他框进行对比,进行iou的计算
当iou的数值大于给定的阈值如0.5,则保留;如果没有大于0.5的框就丢弃掉
当剩下的iou有好几个都是大于0.5的,yolo就会保留最大的作为边界框。这应该就是NMS的过程。
如果画面中会有其他的猫或者狗
不同的猫与猫 不同的狗与狗之间是不会有框的交集的,因此在做iou计算时彼此并没有影响。