[发明专利]一种图像中物体及其关键点的确定方法和装置

说明书

本申请公开了一种图像中物体及其关键点的确定方法和装置，在从待检测图像中提取到该待检测图像对应的图像特征后，直接根据该待检测图像对应的图像特征，确定目标物体在待检测图像上的位置信息以及该目标物体的关键点位置信息。其中，由于目标物体在待检测图像上的位置信息能够准确地表征待检测图像中目标物体的位置，且目标物体的关键点位置信息能够准确地表征待检测图像中目标物体的关键点位置，因而，本申请实施例提供的图像中物体及其关键点的确定方法能够有效地从待检测图像中确定出目标物体的位置信息及其关键点的位置信息，从而能够准确地检测出图像中物体及其关键点。

技术领域

本申请涉及图像处理技术领域，尤其涉及一种图像中物体及其关键点的确定方法和装置。

背景技术

随着车辆的普及，车辆驾驶安全越来越重要。其中，驾驶员的疲劳驾驶是一种能够影响驾驶安全的重要因素。

目前，通常使用车辆的疲劳驾驶检测系统确定驾驶员是否处于疲劳驾驶状态，以便在确定驾驶员处于疲劳驾驶状态时对驾驶员采取相应的提醒措施。其中，为了能够有效地确定出驾驶员是否处于疲劳驾驶状态，需要根据包括驾驶员脸部的图像，进行人脸检测以及人脸关键点检测，以便后续能够利用人脸检测结果以及人脸关键点检测结果确定驾驶员的精神状态。

有上述分析可知，为了提高驾驶安全性，需要准确地检测出图像中人脸以及关键点，以便能够基于该检测所得的人脸及其关键点准确地判断驾驶员的精神状态。然而，如何根据准确地检测出图像中物体(例如，人脸)及其关键点却是一个亟待解决的技术问题。

发明内容

为了解决现有技术中存在的以上技术问题，本申请提供一种图像中物体及其关键点的确定方法和装置，能够有效地确定出图像中物体位置及其关键点位置，从而能够准确地检测出图像中物体及其关键点。

为了实现上述目的，本申请实施例提供的技术方案如下：

本申请实施例提供一种图像中物体及其关键点的确定方法，包括：

对待检测图像进行特征提取，得到所述待检测图像对应的图像特征；

根据所述待检测图像对应的图像特征，确定目标物体在待检测图像上的位置信息以及所述目标物体的关键点位置信息；其中，所述目标物体的关键点用于表征所述目标物体的结构特征。

可选的，所述根据所述待检测图像对应的图像特征，确定目标物体在待检测图像上的位置信息以及所述目标物体的关键点位置信息，具体为：

根据所述待检测图像对应的图像特征，确定目标物体在待检测图像上的位置信息，并根据所述待检测图像对应的图像特征和所述目标物体在待检测图像上的位置信息，确定目标物体的关键点位置信息。

可选的，所述根据所述待检测图像对应的图像特征，确定目标物体在待检测图像上的位置信息，并根据所述待检测图像对应的图像特征和所述目标物体在待检测图像上的位置信息，确定目标物体的关键点位置信息，具体包括：

根据所述待检测图像对应的图像特征，利用预先构建的第一检测模型对目标物体及其关键点进行检测，确定目标物体在待检测图像上的位置信息以及目标物体的关键点位置信息；

其中，所述第一检测模型包括第一物体位置检测网络层和第一关键点位置检测网络层，且所述第一物体位置检测网络层的输出结果是所述第一关键点位置检测网络层的输入数据，且所述第一物体位置检测网络层用于根据所述待检测图像对应的图像特征确定目标物体在待检测图像上的位置信息，所述第一关键点位置检测网络层用于根据所述待检测图像对应的图像特征和所述目标物体在待检测图像上的位置信息，确定目标物体的关键点位置信息。

可选的，所述第一检测模型的训练过程，具体包括：

获取训练图像对应的图像特征、目标物体在训练图像上的实际位置信息以及目标物体的实际关键点位置信息；