GitHub (opens new window)

导言

2021-10-3 About 1 min

# 计算机视觉面临的挑战



# NMS

NMS是大部分深度学习目标检测网络所需要的,大致算法流程为:

  1. 对所有预测框的置信度降序排序
  2. 选出置信度最高的预测框,确认其为正确预测(下次就没有他了,已经被确认了),并计算他与其他预测框的IOU
  3. 根据2中计算的IOU去除重叠度高的,IOU>threshold就删除
  4. 剩下的预测框返回第1步,直到没有剩下的为止