YOLOv3中的非极大值抑制

YOLOv3中的非极大值抑制

YOLOv3中的非极大值抑制

上图中NMS的计算流程:右上角列出了预测框和每个预测框的分数(boxes和scores)。就是定义一个列表keep存放每次选出的分数最大且阈值小(阈值小代表两个预测框没重合,也就是框的不是一个对象)的框。step3中选择boxes[10],boxes[10]要和之前keep中存在的3,5进行iou比较,如果与他俩比较的iou都比较小,说明boxes[10]框中的对象和boxes[3]boxes[5]框中的对象不是同一个对象,所以保留下来。图是NMS后的结果:keep中只有三个框了。

YOLOv3中的非极大值抑制

YOLOv3中的非极大值抑制

YOLOv3中的非极大值抑制