[发明专利]目标检测定位置信度确定方法、装置、电子设备及存储介质

说明书

本申请涉及人工智能技术领域，具体涉及一种目标检测定位置信度确定方法、装置、电子设备及存储介质。该方法包括：获取待检测图像；将所述待检测图像输入到目标检测网络进行目标检测，得到至少一个候选框、每个候选框的第一掩码图以及N个第一特征图；对所述每个候选框的第一掩码图进行扩充，得到所述每个候选框的第二掩码图；从所述N个第一特征图中截取N个第二特征图；将所述每个候选框的第二掩码图与所述N个第二特征图进行拼接，得到所述每个候选框对应的第一目标特征图；根据所述每个候选框对应的第一目标特征图，确定所述每个候选框的定位置信度。本申请实施例有利于使高定位置信度更加精确。

技术领域

本申请涉及人工智能技术领域，具体涉及一种目标检测定位置信度确定方法、装置、电子设备及存储介质。

背景技术

目标检测从复杂图像中分离出目标与背景，并指明每个目标的分类以及所在的区域框位置，是计算机视觉的基础任务之一。目标检测作为一种通用的前处理范式广泛应用于众多领域，如图像分割任务中使用目标检测得到的框信息输入分割网络，图像文本生成任务中使用目标检测得到的感兴趣位置特征输入至编解码器结构，目标追踪任务中使用目标检测获得的框位置进行分析。

目前目标检测方法大多使用分类的置信度作为候选框得分的基准。然而，被量化为候选框的信息中，分类好坏与定位好坏并不相关。为了提高评价候选框的准确度，在二阶段目标检测模型上增加了一个全连接神经网络分支用于评价候选框的定位精度。但是，该评价方法仅能应用到二阶段目标检测模型，并且，对候选框定位精度的评价依赖于之前的池化操作，造成用于评价定位精度的信息比较局限和片面。

因此，现有的目标检测，确定候选框定位精度的方式比较单一，准确度低。

发明内容

本申请实施例提供了一种目标检测定位置信度确定方法、装置、电子设备及存储介质，通过拼接底层特征图，提高对候选框的定位置信度的评价精度。

第一方面，本申请实施例提供一种目标检测定位置信度确定方法，包括：

获取待检测图像；

将所述待检测图像输入到目标检测网络进行目标检测，得到至少一个候选框、每个候选框的第一掩码图以及N个第一特征图，所述N个第一特征图为所述目标检测网络的第一网络层的N个通道上的N个特征图，所述第一网络层位于所述目标检测网络的底部，N为大于1的整数；

对所述每个候选框的第一掩码图进行扩充，得到所述每个候选框的第二掩码图；

从所述N个第一特征图中截取N个第二特征图；

将所述每个候选框的第二掩码图与所述N个第二特征图进行拼接，得到所述每个候选框对应的第一目标特征图，其中，所述第一目标特征图是一个维度为W1*H1*(N+1)的三维矩阵，W1*H1为所述每个候选框的第二掩码图以及每个第二特征图的尺寸；

根据所述每个候选框对应的第一目标特征图，确定所述每个候选框的定位置信度。

第二方面，本申请实施例提供一种目标检测定位置信度确定装置，包括：

获取单元，用于获取待检测图像；

处理单元，用于将所述待检测图像输入到目标检测网络进行目标检测，得到至少一个候选框、每个候选框的第一掩码图以及N个第一特征图，所述N个第一特征图为所述目标检测网络的第一网络层的N个通道上的N个特征图，所述第一网络层位于所述目标检测网络的底部，N为大于1的整数；

对所述每个候选框的第一掩码图进行扩充，得到所述每个候选框的第二掩码图；

从所述N个第一特征图中截取N个第二特征图；