【技术实现步骤摘要】
一种剔除重叠框的目标检测后处理方法
本专利技术属于目标检测
,尤其是一种剔除重叠框的目标检测后处理方法。
技术介绍
计算机视觉是一门研究如何使机器“看”的科学,更进一步的说,就是是指用摄影机和电脑代替人眼对目标进行识别、跟踪和测量等机器视觉,并进一步做图形处理,使电脑处理成为更适合人眼观察或传送给仪器检测的图像。作为一个科学学科,计算机视觉研究相关的理论和技术,试图建立能够从图像或者多维数据中获取‘信息’的人工智能系统。这里所指的信息指Shannon定义的,可以用来帮助做一个“决定”的信息。因为感知可以看作是从感官信号中提取信息,所以计算机视觉也可以看作是研究如何使人工系统从图像或多维数据中“感知”的科学。随着计算机视觉技术的快速发展,目标检测算法在越来越多的场景中得到了应用。现有的目标检测算法在执行前向推断时,大多使用非极大值抑制(NMS)剔除掉重叠度高且预测类别相同的边界框。而一些目标检测算法(如CenterNet)在执行推断时会预测出重叠度高且类别不同的边界框,目前还没有能够有效剔除重叠度高且类别不同的边
【技术保护点】
1.一种剔除重叠框的目标检测后处理方法,其特征在于,包括如下步骤:/n步骤1:使用目标检测算法在测试集上执行推断;/n步骤2:将网络预测的边界框按照不同类别进行分组;/n步骤3:计算每一类别与其他类别所有边界框的交并比,根据交并比与置信度剔除掉重叠度高且类别不同的边界框。/n
【技术特征摘要】
1.一种剔除重叠框的目标检测后处理方法,其特征在于,包括如下步骤:
步骤1:使用目标检测算法在测试集上执行推断;
步骤2:将网络预测的边界框按照不同类别进行分组;
步骤3:计算每一类别与其他类别所有边界框的交并比,根据交并比与置信度剔除掉重叠度高且类别不同的边界框。
2.一种如权利要求1的剔除重叠框的目标检测后处理方法的实现方法,其特征在于,所述步骤1的推断方法包括:搭建深度学习开发环境,加载目标检测算法训练好的模型;在测试集上执行推断,输出所有边界框的置信度、类别、位置坐标。
3.如权利要求2所述的剔除重叠框的目标检测后处理方法的实现方法,其特征在于,所述步骤2的分组方法包括:记类别总数为n,则将边界框...
【专利技术属性】
技术研发人员:廖呈玮,林建就,秦卓奕,
申请(专利权)人:桂林汉璟智能仪器有限公司,
类型:发明
国别省市:广西;45
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。