非极大值抑制(Non-Maximum Suppression,NMS)是目标检测中用于去除冗余检测框的一种重要算法,以下是关于它的详细介绍:
算法原理
- 在目标检测任务中,模型会对每个目标生成多个检测框,这些检测框可能有不同的置信度得分,并且会存在大量重叠的情况。NMS的目的就是在这些检测框中筛选出最具代表性的、置信度最高的检测框,同时抑制那些与它重叠度较高的冗余检测框。
- 算法首先根据检测框的置信度得分进行排序,选择得分最高的检测框作为基准框,然后计算其他检测框与该基准框的交并比(IoU)。如果某个检测框与基准框的IoU大于预先设定的阈值,则认为该检测框是冗余的,将其抑制(即从检测结果中...