[发明专利]施工现场的安全帽佩戴检测方法、装置、设备及存储介质

摘要

本发明公开了一种施工现场的安全帽佩戴检测方法、装置、设备及存储介质，通过根据施工现场图像样本以及包括人体头部位置标注和安全帽类别标注的标注信息训练安全帽佩戴检测模型，将待检测图像输入该安全帽佩戴检测模型，得到待检测图像中的人体头部位置信息以及与人体头部位置信息对应的安全帽类别信息并输出，相比于传统的从背景检测人体再检测头部区域再到安全帽识别的图像处理方法，整个过程是端到端的，更关注人体头部特征和安全帽类别特征进行学习，因此能够适应背景不同的多种施工场景，无需针对各施工场景分别训练模型参数，减小了传统模型泛化能力差的问题，便于及时发现未佩戴安全帽的工人、保障工人安全。

主权项

1.一种施工现场的安全帽佩戴检测方法，其特征在于，包括：根据施工现场图像样本以及与所述施工现场图像样本对应的标注信息训练安全帽佩戴检测模型；将待检测图像输入所述安全帽佩戴检测模型，得到所述待检测图像中的人体头部位置信息以及与所述人体头部位置信息对应的安全帽类别信息；输出所述人体头部位置信息和所述安全帽类别信息；其中，所述标注信息包括所述施工现场图像样本上的人体头部位置标注以及与所述人体头部位置标注对应的安全帽类别标注，安全帽类别具体为佩戴安全帽类别或未佩戴安全帽类别。