目标检测算法中的先验框匹配:YOLO与SSD的差异
在训练过程中,首先要确定训练图片中的ground truth(真实目标)与哪个先验框来进行匹配,与之匹配的先验框所对应的边界框将负责预测它。在YOLO中,ground truth的中心落在哪个单元格,该单元格中与其IOU(定义见图8)最大的边界框负责预测它。但是在SSD中却完全不一样,SSD的先验框与ground truth的匹配原则主要有两点。
在训练时,ground truth与 default boxes按照如下方式进行配对:
首先,寻找与每一个ground truth box有最大IoU的default box,这样就能保证每一个ground truth能与至少一个default box对应起来。
SSD之后又将剩余还没有配对的default box与任意一个ground truth尝试配对,只要两者之间的IoU大于阈值,就进行匹配(SSD 300 阈值为0.5)。换个说法,SSD在匹配时会优先选择与ground truth有最大IoU的default box,但如果有default box没有与任何一个ground truth匹配,SSD会将它们与IoU大于阈值的ground truth进行匹配。这种策略可以保证每一个default box都能被匹配到一个ground truth,从而提高检测精度。
原文地址: https://www.cveoy.top/t/topic/gDZc 著作权归作者所有。请勿转载和采集!