[发明专利]一种目标检测方法及目标检测模型的训练方法、装置

权利要求书

1.一种目标检测方法，包括：

获取待检测图像中目标检测对象的最小外接矩形框；

对所述最小外接矩形框中的图像区域进行前景提取，得到所述目标检测对象的前景像素区域；

确定所述前景像素区域的中心得到目标主体中心；

确定所述前景像素区域的主体框得到目标主体框；

基于所述目标主体框以及所述目标主体中心对所述目标检测对象进行匹配，得到所述待检测图像的检测结果。

2.根据权利要求1所述的方法，其中，所述确定所述前景像素区域的中心得到目标主体中心，包括：

基于所述前景像素区域中像素点的个数，以及各像素点的列坐标和行坐标，计算得到所述目标主体中心。

3.根据权利要求2所述的方法，其中，所述基于所述前景像素区域中像素点的个数，以及各像素点的列坐标和行坐标，计算得到所述目标主体中心，包括：

对各所述像素点的列坐标进行求和，将求和结果与所述像素点的个数的比值作为目标主体中心的列坐标；

对各所述像素点的行坐标进行求和，将求和结果与所述像素点的个数的比值作为目标主体中心的行坐标。

4.根据权利要求1所述的方法，其中，

所述获取待检测图像中目标检测对象的最小外接矩形框，包括：

获取待检测图像中目标检测对象的最小外接矩形框的尺寸信息；

所述确定所述前景像素区域的主体框得到目标主体框，包括：

基于所述最小外接矩形框的尺寸信息，以及所述前景像素区域中像素点的个数，确定所述目标主体框。

5.根据权利要求4所述的方法，其中，所述尺寸信息包含宽度和高度，所述基于所述最小外接矩形框的尺寸信息，以及所述前景像素区域中像素点的个数，确定所述目标主体框，包括：

将所述最小外接矩形框的宽度与目标因子的乘积作为目标主体框的宽度；

将所述最小外接矩形框的高度与目标因子的乘积作为目标主体框的高度；

其中，所述目标因子为：所述前景像素区域中像素点的个数与最小外接矩形框面积比值的二次方根。

6.一种目标检测模型训练方法，包括：

获取样本图像中样本检测对象的样本最小外接矩形框；

对所述样本最小外接矩形框中的图像区域进行前景提取，得到所述样本检测对象的样本前景像素区域；

确定所述样本前景像素区域的中心得到样本主体中心；

确定所述样本前景像素区域的主体框得到样本主体框；

利用目标检测模型，基于所述样本主体框以及所述样本主体中心对所述样本检测对象进行匹配，得到所述样本图像的预测结果；

根据所述样本图像的预测结果及所述样本图像的标注真值，计算当前损失；

根据当前损失调整目标检测模型的训练参数，返回执行：获取样本图像中样本检测对象的样本最小外接矩形框，直至满足预设结束条件，得到训练好的目标检测模型。

7.根据权利要求6所述的方法，其中，所述确定所述样本前景像素区域的中心得到样本主体中心，包括：

基于所述样本前景像素区域中像素点的个数，以及各样本像素点的列坐标和行坐标，计算得到所述样本主体中心。

8.根据权利要求7所述的方法，其中，所述基于所述样本前景像素区域中像素点的个数，以及各样本像素点的列坐标和行坐标，计算得到所述样本主体中心，包括：

对各所述样本像素点的列坐标进行求和，将求和结果与所述样本像素点的个数的比值作为样本主体中心的列坐标；

对各所述样本像素点的行坐标进行求和，将求和结果与所述样本像素点的个数的比值作为样本主体中心的行坐标。