[发明专利]一种安全带检测方法、装置、电子设备及存储介质

权利要求书

1.一种安全带检测方法，其特征在于，所述方法包括：

获取待检测的车窗图像；

将所述车窗图像输入预先训练完成的安全带检测模型，基于所述安全带检测模型确定安全带的两个角点位置信息；

根据所述两个角点的位置信息，确定所述两个角点连接的线段的位置信息，根据所述线段的位置信息确定车窗内人员是否佩戴安全带。

2.如权利要求1所述的方法，其特征在于，所述获取待检测的车窗图像包括：

获取图像采集设备采集的图像，将所述图像输入预先训练完成的车辆检测模型，确定车辆子图像；

将所述车辆子图像输入预先训练完成的车窗检测模型，确定车窗子图像；

根据所述车窗子图像确定第一主驾驶车窗图像和第一副驾驶车窗图像，分别将所述第一主驾驶车窗图像和第一副驾驶车窗图像确定为待检测的车窗图像。

3.如权利要求2所述的方法，其特征在于，所述将所述车窗图像输入预先训练完成的安全带检测模型，基于所述安全带检测模型确定安全带的两个角点位置信息包括：

将第一主驾驶车窗图像输入预先训练完成的主驾驶安全带检测模型，基于所述主驾驶安全带检测模型确定主驾驶安全带的两个角点位置信息；

将第一副驾驶车窗图像输入预先训练完成的副驾驶安全带检测模型，基于所述副驾驶安全带检测模型确定副驾驶安全带的两个角点位置信息。

4.如权利要求3所述的方法，其特征在于，所述主驾驶安全带检测模型的训练过程包括：

针对第一训练集中的每张第二主驾驶车窗图像，将该第二主驾驶车窗图像和该第二主驾驶车窗图像对应的第一标注图像，输入所述主驾驶安全带检测模型，对所述主驾驶安全带检测模型进行训练；其中，在该第二主驾驶车窗图像中标注安全带两个角点的位置信息和安全带中心点的位置信息，将该第二主驾驶车窗图像以垂直中心线进行镜像翻转，得到该第二主驾驶车窗图像对应的第一标注图像，或将该第二主驾驶车窗图像以垂直中心线进行镜像翻转，在翻转后的第二主驾驶车窗图像中标注安全带两个角点的位置信息和安全带中心点的位置信息，得到该第二主驾驶车窗图像对应的第一标注图像。

5.如权利要求3所述的方法，其特征在于，所述副驾驶安全带检测模型的训练过程包括：

针对第二训练集中的每张第二副驾驶车窗图像，将该第二副驾驶车窗图像和该第二副驾驶车窗图像对应的第二标注图像，输入所述副驾驶安全带检测模型，对所述副驾驶安全带检测模型进行训练；其中，在该第二副驾驶车窗图像中标注安全带两个角点的位置信息和安全带中心点的位置信息，得到该第二副驾驶车窗图像对应的第二标注图像。

6.如权利要求1所述的方法，其特征在于，所述根据所述线段的位置信息确定车窗内人员是否佩戴安全带包括：

根据所述线段的位置信息确定所述线段的斜率，判断所述斜率是否在预设的斜率范围内，如果是，确定车窗内人员佩戴安全带，如果否，确定车窗内人员未佩戴安全带；或确定所述车窗内人员所在目标区域，判断所述线段与所述目标区域重合的像素点数量是否大于预设的数量阈值，如果是，确定车窗内人员佩戴安全带，如果否，确定车窗内人员未佩戴安全带。

7.如权利要求6所述的方法，其特征在于，所述确定所述车窗内人员所在目标区域包括：

将所述待检测的车窗图像输入预先训练完成的目标分割模型，基于所述目标分割模型，确定所述车窗内人员所在目标区域。

8.如权利要求1所述的方法，其特征在于，如果确定车窗内人员佩戴安全带，所述方法还包括：

如果待检测的车窗图像为第一主驾驶车窗图像，根据所述两个角点的位置信息、所述待检测的车窗图像的中心像素点的位置信息和所述待检测的车窗图像的长度以及宽度信息，确定两个目标角点的位置信息，将所述两个目标角点构成的线段的位置信息确定为安全带的位置信息；

如果待检测的车窗图像为第一副驾驶车窗图像，将所述两个角点构成的线段的位置信息确定为安全带的位置信息。