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

说明书

本公开提供了一种目标检测方法及目标检测模型的训练方法、装置，涉及人工智能技术领域，尤其涉及深度学习、计算机视觉等技术领域。具体实现方案为：获取待检测图像中目标检测对象的最小外接矩形框；对最小外接矩形框中的图像区域进行前景提取，得到目标检测对象的前景像素区域；确定前景像素区域的中心得到目标主体中心；确定前景像素区域的主体框得到目标主体框；基于目标主体框以及目标主体中心对目标检测对象进行匹配，得到待检测图像的检测结果。本公开能够利用所确定的目标主体中心和目标主体框对待检测图像进行检测。

技术领域

本公开涉及人工智能技术领域，尤其涉及深度学习、计算机视觉等技术领域，具体涉及一种目标检测方法及目标检测模型的训练方法、装置。

背景技术

随着人工智能技术的发展，人工神经网络等深度学习模型在目标检测、对象识别、图像分类等领域得到了广泛的应用。在目标检测的过程中，目标检测框的确定是必不可少的环节。

发明内容

本公开提供了一种目标检测方法及目标检测模型的训练方法、装置。

根据本公开的一方面，提供了一种目标检测方法，包括：

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

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

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

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

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

根据本公开的另一方面，提供了一种目标检测模型训练方法，包括：

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

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

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

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

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

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

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

根据本公开的另一方面，提供了一种目标检测装置，包括：

第一获取模块，用于获取待检测图像中目标检测对象的最小外接矩形框；

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

第一中心确定模块，用于确定所述前景像素区域的中心得到目标主体中心；

第一主体框确定模块，用于确定所述前景像素区域的主体框得到目标主体框；

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

根据本公开的另一方面，提供了一种目标检测模型训练装置，包括：

第二获取模块，用于获取样本图像中样本检测对象的样本最小外接矩形框；