[发明专利]利用空间上下文的目标检测方法、装置和计算机设备

权利要求书

1.一种利用空间上下文的目标检测方法，其特征在于，所述方法包括：

分别对初始图像进行标记、对所述初始图像中的检测目标的位置信息进行至少两次不同的标记，得到检测目标在所述初始图像中的定位信息以及提取所述初始图像中的检测目标的轮廓坐标，得到训练数据；

将所述训练数据输入预先构建的目标检测网络；所述目标检测网络包括：基础特征提取网络、空间上下文特征提取网络以及输出网络；

通过所述基础特征提取网络提取所述训练数据进行特征提取，得到二维图像特征；

通过所述空间上下文特征提取网络提取各个方位上所述二维图像特征对应的上下文特征；

根据所述上下文特征和所述二维图像特征，得到空间上下文特征，利用输出网络根据所述空间上下文特征输出初始目标检测结果；

根据所述定位信息和所述轮廓坐标构建损失函数，利用所述损失函数对所述目标检测网络进行反向训练，得到训练好的目标检测网络；

将待检测图像输入训练好的目标检测网络，输出目标检测结果。

2.根据权利要求1所述的方法，其特征在于，对所述初始图像中的检测目标的位置信息进行至少两次不同的标记，得到检测目标在所述初始图像中的定位信息，包括：

获取检测目标所在左上角的位置信息以及获取检测目标所在右下角的位置信息；

对所述左上角和所述右下角的位置信息进行标记，得到检测目标在所述初始图像中的定位信息。

3.根据权利要求1所述的方法，其特征在于，通过所述空间上下文特征提取网络提取各个方位上所述二维图像特征对应的上下文特征，包括：

将所述二维图像特征输入所述空间上下文特征提取网络，得到各个方位上所述二维图像特征对应的上下文特征；所述空间上下文特征提取网络包括上方位上下文特征提取网络、下方位上下文特征提取网络、左方位上下文特征提取网络以及右方位上下文特征提取网络；其中，所述上方位上下文特征提取网络输出上方位上下文特征，所述下方位上下文特征提取网络输出下方位上下文特征，所述左方位上下文特征提取网络输出左方位上下文特征，所述右方位上下文提取网络输出右方位上下文特征。

4.根据权利要求3所述的方法，其特征在于，根据所述上下文特征和所述二维图像特征，得到空间上下文特征，包括：

将所述上方位上下文特征与所述二维特征图进行融合、所述下方位上下文特征与所述二维特征图进行融合、左方位上下文特征与所述二维特征图进行融合以及右方位上下文特征与所述二维特征图进行融合，均通过卷积以及下采样，得到空间上下文特征。

5.根据权利要求2所述的方法，其特征在于，根据所述定位信息和所述轮廓坐标构建损失函数，包括：

根据所述训练数据输入所述目标检测网络输出的预测定位信息与所述定位信息的偏差作为第一偏差；

根据所述训练数据输入所述目标检测网络输出的预测边框坐标与所述边框坐标的偏差作为第二偏差；

根据所述第一偏差和所述第二偏差，构建损失函数。

6.根据权利要求5所述的方法，其特征在于，根据所述第一偏差和所述第二偏差，构建损失函数，包括：

利用方差投票机制，根据所述第一偏差和所述第二偏差，构建损失函数。

7.根据权利要求1至6任一项所述的方法，其特征在于，所述方法还包括：

对所述训练数据中的初始图像进行数据增强处理。