计算机与现代化Issue(12)：66-71,6.DOI:10.3969/j.issn.1006-2475.2024.12.010
复杂施工场景下的安全帽佩戴检测算法
Safety Helmet Wearing Detection Algorithm for Complex Construction Scenes
摘要
Abstract
In view of complex background interference and foreign object occlusion in the construction scenes,which reduces the accuracy of helmet wearing detection,we propose a safety helmet wearing detection algorithm for complex construction scenes.This paper improves the YOLOv5 algorithm,adding the Coordinate Attention(CA)mechanism,replacing the first two layers in the backbone network using the Stem Block,applying a Decoupled detection Head(DH)structure with the addition of the Coordinate Attention mechanism.At the same time,an additional large-scale feature extraction layer is added.Results on the helmet dataset show that the improved CADH-YOLOv5 algorithm with a mean detection precision of 91.2%can significantly im-prove the performance of safety helmet wearing detection for complex construction scenes,which is superior to similar algo-rithms,and has limited real-time performance.关键词
特征提取/安全帽佩戴检测/坐标注意力机制/解耦检测头/目标检测Key words
feature extraction/safety helmet wearing detection/coordinate attention/decoupled detection heads/object detect分类
信息技术与安全科学引用本文复制引用
刘云海,冯广,吴晓婷,杨群..复杂施工场景下的安全帽佩戴检测算法[J].计算机与现代化,2024,(12):66-71,6.基金项目
国家自然科学基金资助项目(62237001) （62237001）
