基于改进的YOLO_X算法与注意力模块相结合的头盔检测方法

摘要

本发明提供基于改进的YOLO_X算法与注意力模块相结合的头盔检测方法，包括以下步骤：S1、获取人佩戴头盔与未佩戴头盔的图像数据，并进行筛选；S2、对筛选后的一部分图像数据进行重采样和数据增强；S3、将经过数据增强与未数据增强的图像数据汇总成数据集，并对图像数据进行标签标注S4、构建基于改进的YOLO_X算法与注意力模块相结合的头盔检测模型；S5、选取数据集中的一部分图像数据作为训练集训练S4的头盔检测模型并进行参数优化，然后选取数据集中的另一部分图像数据作为测试集放入优化后的头盔检测模型中，输出头盔佩戴预测结果。本发明能提高头盔检测算法的准确度，从而辅助相关部门制定相关决定，减轻交通部门与施工单位的工作量。