[发明专利]一种目标检测方法、装置、设备及存储介质

权利要求书

1.一种目标检测方法，其特征在于，包括：

对待检测图像进行零像素扩展，以将所述待检测图像扩展为正方形图像；

将所述正方形图像输入预先训练的尺寸调整模型中，得到所述尺寸调整模型输出的目标图像，其中，包括：将所述正方形图像输入第一卷积网络，得到所述第一卷积网络输出的第一特征图像，所述第一卷积网络包括两个卷积层，将所述第一特征图像输入第二卷积网络，得到所述第二卷积网络输出的第二特征图像，所述第二卷积网络包括两个可变形卷积层和一个下采样层，将所述正方形图像输入第一下采样层，并将所述第一下采样层的输出与所述第二特征图像相加，得到所述目标图像；其中，所述尺寸调整模型中的卷积网络配置有可变形卷积层和下采样层；

将所述目标图像输入预先训练的目标检测模型中，得到所述目标检测模型输出的检测结果，并根据所述检测结果确定所述待检测图像中的目标对象。

2.根据权利要求1所述的方法，其特征在于，所述将所述第一特征图像输入第二卷积网络，得到所述第二卷积网络输出的第二特征图像，包括：

将所述第一特征图像输入第二下采样层，得到所述第二下采样层输出的第三特征图像；

将所述第三特征图像输入第一可变形卷积层，并将所述第一可变形卷积层的输入与输出相加，得到第四特征图像；

将所述第四特征图像输入第二可变形卷积层，得到所述第二可变形卷积层输出的第二特征图像。

3.根据权利要求1-2任一所述的方法，其特征在于，所述将所述目标图像输入预先训练的目标检测模型中，得到所述目标检测模型输出的检测结果，并根据所述检测结果确定所述待检测图像中的目标对象，包括：

将所述目标图像输入所述目标检测模型，得到所述目标检测模型输出的热图，并确定所述热图中的关键点；

根据所述关键点和所述关键点在所述热图中对应的相邻区域拟合高斯函数，并根据所述高斯函数确定所述热图中的关键点位置；

将所述关键点位置映射到所述待检测图像中，确定所述待检测图像中与所述关键点位置对应的目标关键点，并根据所述目标关键点确定出所述待检测图像中的目标对象。

4.根据权利要求3所述的方法，其特征在于，所述目标检测模型包括主干网络和反卷积网络；

相应的，所述将所述目标图像输入所述目标检测模型，得到所述目标检测模型输出的热图，包括：

将所述目标图像输入所述主干网络，得到所述主干网络输出的特征数据，所述主干网络包括一个卷积层和三个瓶颈层；

将所述特征数据输入所述反卷积网络，得到所述反卷积网络输出的热图，所述反卷积网络包括一个上采样层和三个卷积层。

5.根据权利要求3所述的方法，其特征在于，所述根据所述关键点和所述关键点在所述热图中对应的相邻区域拟合高斯函数，并根据所述高斯函数确定所述热图中的关键点位置，包括：

根据所述关键点的像素坐标，确定所述关键点的相邻区域的像素坐标和分数值；

根据所述关键点的分数值和像素坐标，以及所述相邻区域的像素坐标和分数值进行高斯曲面拟合，得到所述高斯函数；

确定所述高斯函数的中心点坐标作为所述关键点位置。

6.根据权利要求3所述的方法，其特征在于，所述目标检测方法还包括：

获取多个训练样本图像和对应标记的关键点坐标，并将所述训练样本图像扩展为正方形样本图像；

将所述正方形样本图像输入初始的尺寸调整模型，得到初始的尺寸调整模型输出的目标样本图像，并确定所述关键点坐标在所述目标样本图像中的映射坐标；

根据预设的高斯函数和所述映射坐标，将所述目标样本图像转换成对应的高斯热图标签；

将所述目标样本图像输入初始的目标检测模型，根据初始的目标检测模型输出的高斯热图和高斯热图标签，调整所述尺寸调整模型和目标检测模型的参数。