基于改进YOLOX的安全帽佩戴检测算法

扫码查看

原文链接

万方数据
维普

中文摘要：针对安全帽佩戴检测任务中的小目标密集导致的遮挡和难以检测等问题,提出了一种基于YOLOX和transformer的安全帽佩戴检测算法.首先使用融合了CNN和transformer的网络swin Csp进行特征提取,提升模型获取全局上下文信息的性能;然后采用简化的重复加权双向特征金字塔网络BiFPN,使算法更好地融合不同尺度的特征;最后在 neck和YOLO head的连接部分加入CBAM注意力机制,加强安全帽特征与位置信息的关联程度,进一步增强网络的性能和鲁棒性.基于SHWD数据集进行实验,结果表明,算法在hat类上提高了0.99％的精度,在两种类型的召回率上分别有3.16％和1.7％的提升,在mAP方面则有1.79％的提升,有效实现了快速且精度较高的安全帽佩戴检测.

作者：

熊伟、张中天、路鑫、和宝同

展开 >

作者单位：

华北电力大学计算机系河北保定 071000

关键词：

深度学习目标检测特征融合注意力机制安全帽识别

出版年：

2024

DOI：

10.3969/j.issn.1672-9528.2024.02.005

信息技术与信息化

山东电子学会

信息技术与信息化

影响因子：0.29

ISSN：1672-9528

年,卷(期)：2024.(2)

参考文献量15