一种基于改进YOLOV5模型的安全帽佩戴检测方法

摘要

本发明公开了一种基于改进YOLOV5模型的安全帽佩戴检测方法，方法包括：从安全帽佩戴图像数据集中随机选取图像进行图像数据增强，得到数据增强后的图像，输入至改进YOLOV5模型中进行训练，得到训练后的改进YOLOV5模型；改进YOLOV5模型包括：在特征提取部分嵌入倒置残差模块与倒置残差注意力模块进行图像特征提取；在特征融合部分设计多尺度特征融合模块进行特征融合，并生成四个不同感受野的检测头；优化预测框回归损失函数；最后将待检测图像输入至训练后的改进YOLOV5模型，得到相关人员是否佩戴安全帽的检测结果；本发明有效改善了施工现场视频监控图像中小目标漏检、误检的问题，提高了安全帽佩戴检测精度。