NMS算法步骤

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

NMS算法步骤如下:

for object in all object:
(1)获取当前目标类别下所有bounding box的信息。
(2)将bounding box按照confidence从高到低排序,并记录当前confidence最大的bounding box。
(3)计算最大confidence对应的bounding box与剩下所有bounding box的IoU,移除所有大于IoU阈值的bounding box。
(4)对剩下的bounding box循环执行(2)和(3),直到所有的bounding box满足要求(即不再移除bounding box)。

猜你喜欢

转载自blog.csdn.net/Roaddd/article/details/111240555
NMS