其他分享
首页 > 其他分享> > 『MTCNN』NMS代码实现

『MTCNN』NMS代码实现

作者:互联网

1.练习_从大到小排序

import numpy as np

a=np.array([[5,3],[3,2],[1,6]])
print(a)
#取出第二列数据
d=a[:,1]
print(d)
#按照第二列数据从小到大排序。排序规则为:按照第二列数从小到大对应的索引进行排序
#1.取出第二列数据从小到大对应的索引值。
index=a[:,1].argsort()
print(index)
#2.根据索引排序
print(a[index])#将索引放到a中即可

#从大到小排序。对索引添加符号即可。-a[:,1].argsort()
print(a[-a[:,1].argsort()])

打印结果:

[[5 3]
 [3 2]
 [1 6]]
[3 2 6]
[1 0 2]
[[3 2]
 [5 3]
 [1 6]]
[[1 6]
 [5 3]
 [3 2]]

2.练习_where函数

import numpy as  np

a=np.array([8,7,6,3,1,9])

print(np.where(a<5))#返回索引
print(a[np.where(a<5)])#返回索引值

打印结果:

(array([3, 4], dtype=int64),)
[3 1]

3.练习_stack函数

"""把数组组装成矩阵"""
import numpy as np

a=np.array([1,2])
b=np.array([3,4])
c=np.array([5,6])

ls=[]

ls.append(a)
ls.append(b)
ls.append(c)
print(ls)

#组装
print(np.stack(ls))

打印结果:

[array([1, 2]), array([3, 4]), array([5, 6])]
[[1 2]
 [3 4]
 [5 6]]

4.NMS代码

import numpy as np

"""IOU"""
def iou(box,boxes,isMin=False):#框的格式定义为:[X1,Y1,X2,Y2,C]。一个框和一堆框做比较。为了区分交集是与并集作比较还是和最小面积做比较,先将最小面积赋予默认值0.
    #计算每个框的面积
    box_area=(box[2]-box[0])*(box[3]-box[1])#先计算box的面积。一个框的面积计算:(X2-X1)*(Y2-Y1)。索引拿到坐标值:(box[2]-box[0])*(box[3]-box[1])
    boxes_area=(boxes[:,2]-boxes[:,0])*(boxes[:,3]-boxes[:,1])#一堆框的格式:[[],[],[],[],[],...]

    #计算交集面积
    xx1=np.maximum(box[0],boxes[:,0])#左上角X。交集左上角点坐标:两个相交原框中左上角X和Y各自取较大值,作为交集左上角坐标。用于比较的框的左上角x值:box[0];被比较的框的左上角x值:boxes[0]。去两者较大值。
    yy1=np.maximum(box[1],boxes[:,1])#同理.左上角Y。
    xx2 = np.minimum(box[2], boxes[:, 2])  # 同理.右下角X。
    yy2 = np.minimum(box[3], boxes[:, 3])  # 同理.右下角Y。

    #判断是否有交集
    w=np.maximum(0,xx2-xx1)#当xx2-xx1的值为负值时,表示没有交集,将没有交集的结果变成0即可。使用maximum函数取较大值。
    h=np.maximum(0,yy2-yy1)#同理。

    #正式计算交集面积
    inter=w*h

    if isMin:#如果isMin为True,表示除以最小面积。
        over=np.true_divide(inter,np.minimum(box_area,boxes_area))#true_divide:除法。isMin为True时,除以最小面积。如何得到最小面积呢?比较box_area和boxes_area,取最小值就可得到最小面积。
    else:#否则,除以并集面积。
        over = np.true_divide(inter, (box_area+boxes_area-inter))#两个矩形面积相加减去交集面积

    return over

"""NMS"""
def nms(boxes,thresh=0.3,isMin=False):#需要所有框、阈值、最小面积(传到IOU中,因为IOU是放在NMS里边计算的)
    #根据置信度从大到小排序。
    _boxes=boxes[(-boxes[:,4]).argsort()]       #框的格式定义为:[[X1,Y1,X2,Y2,C],[],[],[],[],...]

    #保留剩余的框
    r_boxes=[]
    #取出第一个框。因为要取很多次,使用循环。(重点)
    while _boxes.shape[0]>1:#循环取出第一个框(shape[0]),当循环过程中取出的维度大于1,说明有框;当维度小于1,表示框已经取完,循环结束。
        #取出第一个框
        a_box=_boxes[0]
        #取出剩余框
        b_boxes=_boxes[1:]
        #保留第一个框
        r_boxes.append(a_box)

        #比较IOU后,保留阈值较小的值
        index=np.where(iou(a_box,b_boxes,isMin)<thresh)#将iou于阈值作比较:iou(a_box,b_boxes,isMin)<thresh,如果iou小于阈值,保留。使用np.where,当小于为True。
        _boxes=b_boxes[index]
    if _boxes.shape[0]>0:
        r_boxes.append(_boxes[0])

    #组装为矩阵
    return np.stack(r_boxes)

if __name__ == '__main__':
    bs=np.array([[2,2,30,30,40],[3,3,25,25,60],[18,18,27,27,15]])
    print(nms(bs))

打印结果:

[[ 3  3 25 25 60]
 [18 18 27 27 15]]
Clb@Xzh 发布了15 篇原创文章 · 获赞 23 · 访问量 1187 私信 关注

标签:box,NMS,area,代码,MTCNN,print,boxes,np,array
来源: https://blog.csdn.net/sinat_39783664/article/details/104142808