YOLOv5算法改进(19)— 手把手教你去更换NMS(DIoU-NMS/CIoU-NMS/EIoU-NMS/GIoU-NMS/SIoU-NMS)

YOLOv5算法改进(19)— 手把手教你去更换NMS(DIoU-NMS/CIoU-NMS/EIoU-NMS/GIoU-NMS/SIoU-NMS)_第1张图片

前言:Hello大家好,我是小哥谈。YOLOv5中的NMS指非极大值抑制(Non-Maximum Suppression),它是一种用于目标检测算法中的后处理技术。在检测到多个重叠的边界框时,NMS可以帮助选择最佳的边界框。NMS的工作原理是首先根据预测边界框的置信度对它们进行排序,然后从置信度最高的边界框开始遍历,将与当前边界框的重叠度(通常使用IoU,即交并比)大于某个阈值的边界框移除。这样可以保留置信度最高的边界框,并且消除冗余的边界框。本节课就简单介绍一下常见的NMS并重点讲解如何去更换NMS!  

YOLOv5算法改进(19)— 手把手教你去更换NMS(DIoU-NMS/CIoU-NMS/EIoU-NMS/GIoU-NMS/SIoU-NMS)_第2张图片 前期回顾:

            

你可能感兴趣的:(YOLOv5:从入门到实战,YOLO,计算机视觉,人工智能,目标检测,机器学习)