目标检测后处理方法、装置、电子设备、车辆及存储介质制造方法及图纸

本发明专利技术提供一种目标检测后处理方法、装置、电子设备、车辆及存储介质，通过获取针对待检测图像的原始中心点热力图预测结果和目标框尺寸预测结果；将原始中心点热力图预测结果中各目标点按照置信度进行分类；根据置信度大于预设置信度阈值的第一目标点集合得到第一目标框集合；根据置信度不大于预设置信度阈值的第二目标点集合得到第二目标框集合；根据第一目标框集合和第二目标框集合得到针对该待检测图像的目标预测结果，在非极大值抑制操作前先对目标点按照置信度进行分类，对数据进行筛选，避免非极大值抑制操作数据量大、耗时长、处理结果不准确的问题，提高目标检测效率和准确性。确性。确性。

【技术实现步骤摘要】
目标检测后处理方法、装置、电子设备、车辆及存储介质


[0001]本专利技术涉及图像识别
，尤其涉及一种目标检测后处理方法、装置、电子设备、车辆及存储介质。

技术介绍

[0002]目标检测是计算机视觉领域的基本任务之一，目标检测一般是在图像中定位目标物体并赋予目标物体相应的标签。目标检测模型的后处理阶段包含两个步骤，首先滤除类别得分低于阈值的预测结果，之后过滤掉重叠的矩形框，得到最终的目标检测结果。传统的目标检测模型的后处理方法直接在类别筛选后，使用传统NMS(Non
‑
Maximum Suppression，非极大值抑制)算法进行过滤操作，导致NMS处理数据量大，耗时长，处理结果不准确。

技术实现思路

[0003]本专利技术提供一种目标检测后处理方法、装置、电子设备、车辆及存储介质，用以解决传统目标检测后处理方法中NMS处理数据量大、耗时长，处理结果不准确导致最终检测结果不准确的缺陷。
[0004]本专利技术提供一种目标检测后处理方法，包括：
[0005]获取针对待检测图像的原始中心点热力本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种目标检测后处理方法，其特征在于，包括：获取针对待检测图像的原始中心点热力图预测结果和目标框尺寸预测结果；将所述原始中心点热力图预测结果中各目标点按照置信度进行分类；根据置信度大于预设置信度阈值的第一目标点集合得到第一目标框集合；根据置信度不大于预设置信度阈值的第二目标点集合得到第二目标框集合；根据所述第一目标框集合和所述第二目标框集合得到针对所述待检测图像的目标预测结果。2.根据权利要求1所述的目标检测后处理方法，其特征在于，所述将所述原始中心点热力图预测结果中各目标点按照置信度进行分类，包括：对原始中心点热力图预测结果进行最大值池化过滤操作；将置信度分数大于置信度阈值的目标点集合作为置信度大于预设置信度阈值的第一目标点集合，置信度分数不大于置信度阈值的目标点集合作为置信度不大于预设置信度阈值的第二目标点集合。3.根据权利要求2所述的目标检测后处理方法，其特征在于，所述根据置信度大于预设置信度阈值的目标点集合得到第一目标框集合，包括：将置信度大于预设置信度阈值的第一目标点集合和所述目标框尺寸预测结果进行融合得到第一目标框集合。4.根据权利要求3所述的目标检测后处理方法，其特征在于，所述根据置信度不大于预设置信度阈值的第二目标点集合得到第二目标预测结果，包括：将置信度不大于预设置信度阈值的第二目标点集合和所述目标框尺寸预测结果进行融合得到第三目标框集合；将所述第三目标框集合与所述第一目标框集合进行合并得到所有预测目标框；对所有预测目标框中属于预测类别的目标框进行缩放操作；对缩放操作后目标框进行非极大值抑制操作；对所述非极大值抑制操作后的目标框进行反缩放操作，得到第二目标框集合。5.根据权利要求4所述的目标检测后...

【专利技术属性】
技术研发人员：陆强，
申请(专利权)人：际络科技上海有限公司，
类型：发明
国别省市：