yolov8 soft-nms
时间: 2023-11-15 14:57:42 浏览: 361
YOLOV8轻量版模块
5星 · 资源好评率100%
Yolov8是一种目标检测算法,它引入了soft-NMS。Soft-NMS是一种非极大值抑制(NMS)的改进方法,它不是直接删除所有与最高得分框的IOU大于阈值的框,而是通过降低这些框的置信度来保留更多的框。在Yolov8中,soft-NMS的实现是通过修改ultralytics/yolo/utils/ops.py文件中的代码来完成的。具体来说,soft-NMS的做法是:对于待处理框bi和当前得分最高框M,如果它们的IOU越大,那么bi的得分si就会下降得越厉害。在Yolov8中,第一处修改点是在torchvision.ops.nms函数中加入了soft-NMS的处理。
阅读全文