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计算时彼此并没有影响。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐