如何用Python编写非极大值抑制（NMS）算法的代码实现？

本文共计843个文字，预计阅读时间需要4分钟。

NMS算法在目标检测中应用广泛，其原理是搜索局部极大值，抑制极大值元素。作用是当算法对某个目标进行检测时，抑制其他非极大值元素。

NMS 算法在目标检测，目标定位领域有较广泛的应用。

算法原理

非极大值抑制算法（Non-maximum suppression, NMS）的本质是搜索局部极大值，抑制非极大值元素。

算法的作用

当算法对一个目标产生了多个候选框的时候，选择 score 最高的框，并抑制其他对于改目标的候选框

适用场景

一幅图中有多个目标（如果只有一个目标，那么直接取 score 最高的候选框即可）。

算法的输入

算法对一幅图产生的所有的候选框，以及每个框对应的 score （可以用一个 5 维数组 dets 表示，前 4 维表示四个角的坐标，第 5 维表示分数），阈值 thresh。

算法的输出

正确的候选框组（dets 的一个子集）。