一种目标检测模型预测框处理方法及装置
基本信息
申请号 | CN201910753422.9 | 申请日 | - |
公开(公告)号 | CN110472640B | 公开(公告)日 | 2022-03-15 |
申请公布号 | CN110472640B | 申请公布日 | 2022-03-15 |
分类号 | G06V10/44(2022.01)I;G06K9/62(2022.01)I;G06N3/04(2006.01)I;G06V10/75(2022.01)I;G06V10/74(2022.01)I;CN 110084173 A,2019.08.02;CN 109934121 A,2019.06.25;CN 110008953 A,2019.07.12;CN 107871134 A,2018.04.03;CN 107784288 A,2018.03.09;CN 109472193 A,2019.03.15;CN 108830205 A,2018.11.16;CN 108229649 A,2018.06.29;CN 109685152 A,2019.04.26;CN 107481265 A,2017.12.15;CN 104463138 A,2015.03.25 胡孟.目标检测番外篇(3)_NMS.《https://zhuanlan.zhihu.com/p/49481833》.2018,;陈珊 等.基于多尺度特征的乳腺肿块检测方法研究.《智能计算机与应用》.2019,第9卷(第1期),;Yihui He 等.Bounding Box Regression with Uncertainty for Accurate Object Detection.《arXiv》.2019,;Borui Jiang 等.Acquisition of Localization Confidence for Accurate Object Detection.《arXiv》.2018,;张强 等.一种改进约束条件的简化非极大值抑制.《中国科学技术大学学报》.2016,第46卷(第1期),;陈金辉 等.行人检测中非极大值抑制算法的改进.《华东理工大学学报(自然科学版)》.2015,第41卷(第3期),;Songtao Liu 等.Adaptive NMS: Refining Pedestrian Detection in a Crowd.《arXiv》.2019, | 分类 | 计算;推算;计数; |
发明人 | 王子彤;姜凯;秦刚 | 申请(专利权)人 | 山东浪潮数字能源科技有限公司 |
代理机构 | 济南信达专利事务所有限公司 | 代理人 | 孙园园 |
地址 | 250100 山东省济南市高新浪潮路1036号S02号楼 | ||
法律状态 | - |
摘要
摘要 | 本发明公开了一种目标检测模型预测框处理方法及装置,属于图像识别目标检测领域,本发明要解决的技术问题为如何消除由于预测框遮挡造成的同一种类不同个体的预测框被错误剔除现象,提高目标检测算法结果的精确度,技术方案为:该方法步骤如下:S1、经过多层卷积神经网络,目标检测模型给出待预测所有种类的全部预测框坐标及置信度;S2、选择属于同一种类的预测框,对置信度进行由大到小排序,暂存于临时输出框集合;S3、选取置信度最大的框,保存至最终输出框集合;S4、判断同一种类是否完成;S5、选取下一排序置信度的框,计算当前框与上一框的坐标位置交并比IOU,将坐标位置交并比IOU与阀值相比较。本发明还公开了目标检测模型预测框处理装置。 |
