在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版权协议,转载请附上原文出处链接和本声明。
在YOLOv5源码中,原作者使用的是普通的nms,还有附加可开启的加权nms(merge-NMS),我们可以将nms替换为ciou_nms等其他的变种nms。
具体操作如下:
Pytorch机器学习(八)—— YOLOV5中NMS非极大值抑制与DIOU-NMS等改进_lzzzzzzm的博客-CSDN博客_yolov5的nms