[发明专利]工帽佩戴检测方法、系统、存储介质及计算机设备

权利要求书

1.一种工帽佩戴检测方法，其特征在于，包括步骤：

获取待测图像；

对所述待测图像进行人体特征检测，确定所述待测图像中对应所述人体特征的第一ROI区域；

对所述第一ROI区域进行人体头部检测，确定所述待测图像中对应人头特征的第二ROI区域；

将所述第二ROI区域输入至预制的工帽分类模型，获取所述工帽分类模型输出的分类识别结果；

根据所述分类识别结果，生成所述待测图像对应的检测结果信息。

2.根据权利要求1所述的工帽佩戴检测方法，其特征在于，所述对所述待测图像进行人体特征检测，确定所述待测图像中对应所述人体特征的第一ROI区域的步骤之前还包括：

根据预设的训练集对人体检测模型、人头检测模型以及所述工帽分类模型进行训练；其中，所述训练集包括有佩戴工帽特征的第一图像样本和未佩戴工帽特征的第二图像样本；

所述对所述待测图像进行人体特征检测，确定所述待测图像中对应所述人体特征的第一ROI区域的步骤具体包括：

通过所述人体检测模型对所述待测图像进行人体检测，以获得所述待测图像中对应人体特征的第一ROI区域；

所述对所述第一ROI区域进行人体头部检测，确定所述待测图像中对应人头特征的第二ROI区域的步骤具体包括：

通过所述人头检测模型对所述第一ROI区域进行人头检测，以获得所述待测图像中对应人头特征的第二ROI区域。

3.根据权利要求2所述的工帽佩戴检测方法，其特征在于，所述通过所述人体检测模型对所述待测图像进行人体检测，以获得所述待测图像中对应人体特征的第一ROI区域的步骤具体包括：

将所述待测图像输入至所述人体检测模型；

根据所述人体检测模型输出的第一检测结果，确定所述待测图像中对应人体特征的第一ROI区域。

4.根据权利要求2所述的工帽佩戴检测方法，其特征在于，所述通过所述人头检测模型对所述第一ROI区域进行人头检测，以获得所述待测图像中对应人头特征的第二ROI区域的步骤具体包括：

将所述第一ROI区域输入至所述人头检测模型；

根据所述人头检测模型输出的第二检测结果，确定所述第一ROI区域中对应人头特征的第二ROI区域。

5.根据权利要求4所述的工帽佩戴检测方法，其特征在于，所述根据所述人头检测模型输出的第二检测结果，确定所述第一ROI区域中对应人头特征的第二ROI区域的步骤具体包括：

遍历所述第二检测结果，以获取符合预设头部尺寸的所述第二ROI区域。

6.根据权利要求2所述的工帽佩戴检测方法，其特征在于，所述根据预设的训练集对人体检测模型、人头检测模型以及所述工帽分类模型进行训练的步骤具体包括：

对所述训练集的图像数据进行图像裁切，以分别获得对应人体区域的第一样本数据集、对应人头区域的第二样本数据集以及对应工帽区域的第三样本数据集；

根据所述第一样本数据集、所述第二样本数据集以及所述第三样本数据集分别对所述人体检测模型、所述人头检测模型以及所述工帽分类模型进行训练。

7.根据权利要求1所述的工帽佩戴检测方法，其特征在于，所述根据所述分类识别结果，生成所述待测图像对应的检测结果信息的步骤具体包括：

根据所述分类识别结果确定佩戴工帽或未佩戴工帽的目标人头区域；

获取所述目标人头区域在所述待测图像上的人头位置信息；

根据所述人头位置信息和所述分类识别结果，生成所述待测图像对应的所述检测结果信息。