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

权利要求书

1.一种目标检测定位置信度确定方法，其特征在于，包括：

获取待检测图像；

将所述待检测图像输入到目标检测网络进行目标检测，得到至少一个候选框、每个候选框的第一掩码图以及N个第一特征图，所述N个第一特征图为所述目标检测网络的第一网络层的N个通道上的N个特征图，所述第一网络层位于所述目标检测网络的底部，具体为：所述第一网络层在所述目标检测网络中的位置满足以下公式：其中，k表示所述第一网络层在所述目标检测网络包含的多个网络层中的位置，k₀为预设的超参，W₁和H₁分别为所述每个候选框的第二掩码图的宽和高，W₀和H₀为所述目标检测网络所规定的尺寸，N为大于1的整数；

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

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

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

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

2.根据权利要求1所述的方法，其特征在于，在从N个第一特征图中截取N个第二特征图之前，所述方法还包括：

获取所述目标检测网络所规定的尺寸；

根据所述每个候选框的第二掩码图的尺寸、所述目标检测网络所规定的尺寸以及预设的超参，确定所述目标检测网络的第一网络层。

3.根据权利要求1或2所述的方法，其特征在于，所述根据所述每个候选框对应的第一目标特征图，确定所述每个候选框的定位置信度，包括：

对所述每个候选框对应的第一目标特征图进行特征提取，得到所述每个候选框对应的第二目标特征图；

通过两个卷积核尺寸为1*1的卷积层先后对所述每个候选框对应的第二目标特征图进行卷积处理，得到所述每个候选框对应的特征向量；

将所述每个候选框对应的特征向量进行归一化处理，得到所述每个候选框的定位置信度。

4.根据权利要求1所述的方法，其特征在于，在根据所述目标特征图，确定所述每个候选框的定位置信度之后，所述方法还包括：

获取所述每个候选框的分类置信度；

根据所述每个候选框的分类置信度以及所述每个候选框的定位置信度，确定所述每个候选框的目标置信度；

根据所述每个候选框的目标置信度，对所述至少一个候选框进行非极大值抑制，得到所述至少一个候选框中的目标候选框。

5.根据权利要求4所述的方法，其特征在于，所述待检测图像包括医学图像。

6.根据权利要求5所述的方法，其特征在于，所述将所述待检测图像输入到目标检测网络进行目标检测，得到至少一个候选框以及所述至少一个候选框中每个候选框的第一掩码图，包括：

将所述医学图像输入到目标检测网络，得到所述医学图像的第三特征图；

获取与所述医学图像所属的医学领域对应的医学知识图谱；

将所述医学知识图谱进行向量化，得到与所述医学知识图谱对应的第一特征向量；

将所述第一特征向量进行多次复制与组合，得到第一三维矩阵，其中，所述第一三维矩阵中每层的数据相同，且所述第一三维矩阵的每层的尺寸与所述第三特征图的尺寸相同；

将所述第一三维矩阵与所述医学图像的第三特征图进行拼接，得到第四特征图，根据所述第四特征图对所述医学图像进行目标检测，得到所述医学图像中与目标对应的至少一个候选框以及所述每个候选框对应的第一掩码图。