YOLOv5检测框的保留

Question

怎么保留不同类别中置信值较高的检测框？这个里面最大值抑制设定又会抑制较低执行度的检测框（减少了检测效果）。
能不能保留不同类中检测置信度高的检测框

Answer 1

引用 皆我百晓生 小程序回复内容作答：

要保留不同类别中置信度较高的检测框，可以使用最大值抑制（NMS）算法。NMS算法会根据置信度对检测框进行排序，然后遍历每个框，计算其与其他框的重叠度（比如IoU），如果重叠度大于设定的阈值，则将该框删除，直到遍历完所有框。

YOLOv5中使用了一种改进的NMS算法，即greedy-NMS。它与传统的NMS算法不同之处在于，它只在置信度最高的框上执行NMS操作，对其他类别的框保持不变。这样可以减少对置信度较低的框的抑制，提高检测效果。

如果想保留不同类别中检测置信度高的检测框，可以根据需要调整NMS算法的阈值，减小阈值则保留更多的框，增大阈值则保留更少的框。可以根据实际情况进行试验，找到一个适合的阈值。