非极大值抑制,删选矩形框,nonmaximum suppression, python,代码实现

#encoding = utf-8

import numpy as np
"""[x1,x2,y1,y2,c]下面的mat数组,各元素的意义:(x1,y1)为矩形左上角顶点的坐标,(x2,y2)为矩形右下角顶点的坐标,c为置信度即confidence的首字母"""
mat = np.array([[20,60,300,500,0.98],[40,80,400,600,0.97],[500,600,400,500,0.72],[125,456,495,550,0.65],[90,160,240,400,0.55],[210,600,400,400,0.75],[220,260,240,300,0.95]])

def py_nms(mat):
    """non maximum suppression """
                         
    mat_all_4_argsort = mat[:,4].argsort()     # 取出mat的第5列并由小到大排列


    keep = np.array([])
    keep = np.append(keep, mat[mat_all_4_argsort[-1]], axis=0) # 向keep中添加最大置信度的一组数据
    keep = keep.reshape(1,-1)                                  # 把keep 转化为二维数组

    for item in mat:
        index_max = mat_all_4_argsort[-1]


        s0 = (mat[index_max][1] - mat[index_max][0])*(mat[index_max][3] - mat[index_max][2])

        x1 = np.maximum(mat[index_max][0],item[0])                     #取两矩形最小x的最大值
        x2 = np.minimum(mat[index_max][1],item[1])                     #取两矩形最大x的最小值
        y1 = np.maximum(mat[index_max][2],item[2])                     #取两矩形最小y的最大值
        y2 = np.minimum(mat[index_max][3],item[3])                     #取两矩形最大y的最小值
        s1 = (x2-x1)*(y2-y1)                                           # 两矩形的相交面积

        if ((s1 / s0) < 0.35) and ((s1 / s0) > 0) :                  # 同时满足两项要求 大于0是必须的,意在除去两个矩形不相交的情况
            keep = np.append(keep,item.reshape(1,-1), axis=0)
    return keep

print('----------------------------')

res = py_nms(mat)
print(res)
已标记关键词 清除标记
【为什么还需要学习C++?】 你是否接触很多语言,但从来没有了解过编程语言的本质? 你是否想成为一名资深开发人员,想开发别人做不了的高性能程序? 你是否经常想要窥探大型企业级开发工程的思路,但苦于没有基础只能望洋兴叹?   那么C++就是你个人能力提升,职业之路进阶的不二之。 【课程特色】 1.课程共19大章节,239课时内容,涵盖数据结构、函数、类、指针、标准库全部知识体系。 2.带你从知识与思想的层面从0构建C++知识框架,分析大型项目实践思路,为你打下坚实的基础。 3.李宁老师结合4大国外顶级C++著作的精华为大家推出的《征服C++11》课程。 【学完后我将达到什么水平?】 1.对C++的各个知识能够熟练配置、开发、部署; 2.吊打一切关于C++的笔试面试题; 3.面向物联网的“嵌入式”和面向大型化的“分布式”开发,掌握职业钥匙,把握行业先机。 【面向人群】 1.希望一站式快速入门的C++初学者; 2.希望快速学习 C++、掌握编程要义、修炼内功的开发者; 3.有志于挑战更高级的开发项目,成为资深开发的工程师。 【课程设计】 本课程包含3大模块 基础篇 本篇主要讲解c++的基础概念,包含数据类型、运算符等基本语法,数组、指针、字符串等基本词法,循环、函数、类等基本句法等。 进阶篇 本篇主要讲解编程中常用的一些技能,包含类的高级技术、类的继承、编译链接和命名空间等。 提升篇: 本篇可以帮助学员更加高效的进行c++开发,其中包含类型转换、文件操作、异常处理、代码重用等内容。
©️2020 CSDN 皮肤主题: 数字20 设计师:CSDN官方博客 返回首页