如何解决目标检测两个物体重叠问题

非极大值抑制(Non-Maximum Suppression)

IoU:intersection-over-union,即两个边界框的交集部分除以它们的并集。

非极大值抑制的流程如下:

  1. 根据置信度得分进行排序
  2. 选择置信度最高的比边界框添加到最终输出列表中,将其从边界框列表中删除
  3. 计算所有边界框的面积
  4. 计算置信度最高的边界框与其它候选框的IoU。
  5. 删除IoU大于阈值的边界框
  6. 重复上述过程,直至边界框列表为空

所以一般具体解决办法就是减小 IoU threshold (IoU 阈值)

你可能感兴趣的:(如何解决目标检测两个物体重叠问题)