YOLOv5改进实战 | 更换NMS之Soft-NMS、DIoU_NMS篇


在这里插入图片描述


前言

☘️ NMS(非极大值抑制)常用于目标检测任务中,用于消除重叠的边界框,保留最具代表性的边界框。NMS算法并不是一种后处理,而是一种在目标检测算法中常用的技术。在目标检测过程中,通常会生成一系列候选边界框,而NMS算法会对这些候选边界框进行筛选,去除冗余的边界框,从而提高目标检测的准确性和效率。因此,NMS可以被视为一种在目标检测后处理阶段使用的技术。

目录

  • 一、添加NMS模块
  • 二、Soft-NMS
    • 代码实现
  • 二、DIoU-NMS
    • 代码实现
  • 三、CIoU-NMS

一、添加NMS模块

  • 修改general.py,增加Soft-NMS模块,代码如下:

你可能感兴趣的:(YOLO改进系列,#,YOLOv5改进系列,YOLO,目标跟踪,人工智能)