python实现NMS算法

永兴呵呵哒

于 2019-07-17 21:30:50 发布

阅读量412

点赞数

本文链接：https://blog.csdn.net/u014106566/article/details/96313237

版权

本文详细介绍了如何使用Python实现非极大值抑制（Non-Maximum Suppression, NMS）算法，该算法常用于目标检测中，以消除重叠的检测框。通过理解NMS的基本原理，我们将一步步构建NMS函数，并给出实际应用的例子。" 7726967,55669,C# 编译与使用DLL详解,"['c#', 'dll', 'library', 'visual studio', 'programming']

摘要由CSDN通过智能技术生成

参考博客：https://blog.csdn.net/hongxingabc/article/details/78996407

import numpy as np

thresh = 0.6                                             #设置阈值
nums = np.array([[100,100,200,200,0.8],
                 [300,300,400,400,0.75],
                 [101,101,202,202,0.78],
                 [135,135,345,345,0.65],
                 [400,230,567,345,0.9],
                 [12,12,98,98,0.9]])                    #建立数组

x1 = nums[:, 0]                                         #得到所有边框的第一个点的x坐标
y1 = nums[:, 1]  
x2 = nums[:, 2]  
y2 = nums[:, 3]  
scores = nums[:, 4]                                     #得到所有边框的置信度分数

areas = (x2 - x1 + 1) * (y2 - y1 + 1)                   #得到每个边框的面积
order = scores.argsort()[::-1]                          #分数从大到小排列，取index
keep = []                                               #keep为最后保留的边框
while order.size > 0:                                
    i = order[0]