YOLOv3中的非极大值抑制

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