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