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

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

具体操作如下:

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


版权声明:本文为weixin_42182534原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。