YOLOV5中NMS的理解和替换(CIOU_NMS)

在YOLOv5源码中,原作者使用的是普通的nms,还有附加可开启的加权nms(merge-NMS),我们可以将nms替换为ciou_nms等其他的变种nms。

具体操作如下:

Pytorch机器学习(八)—— YOLOV5中NMS非极大值抑制与DIOU-NMS等改进_lzzzzzzm的博客-CSDN博客_yolov5的nms

你可能感兴趣的:(深度学习,计算机视觉,目标检测,nms)