第42卷第5期2023年10月沈阳理工大学学报JournalofShenyangLigongUniversityVol42No5Oct2023收稿日期:2022-09-21基金项目:辽宁省教育厅高等学校基本科研项目(LJKZ0241)作者简介:陈扬(1997—)ꎬ女ꎬ硕士研究生ꎮ通信作者:吕艳辉(1971—)ꎬ女ꎬ教授ꎬ博士ꎬ研究方向为计算机视觉、人工智能ꎮ文章编号:1003-1251(2023)05-0011-07基于改进YOLOv5s的头盔佩戴检测算法陈扬ꎬ吕艳辉(沈阳理工大学信息科学与工程学院ꎬ沈阳110159)摘要:交通法规规定电动车驾驶人驾车时需要佩戴安全头盔ꎬ常用检测算法针对安全头盔这类小目标进行检测时存在漏检的问题ꎮ鉴于此ꎬ提出一种基于改进YOLOv5s的电动车驾驶人头盔佩戴检测算法ꎬ简称为HWD ̄YOLOv5s算法ꎮ该算法以深度学习框架YOLOv5s为基础ꎬ改进原始模型特征提取部分的下采样方法和特征融合方法ꎬ并修改边框损失函数GIOU的计算方法ꎮ通过多场景下数据采集获得11370张图片以制作安全头盔数据集ꎬ并在自制数据集上采用HWD ̄YOLOv5s算法及其他主流算法进行小目标检测的对比实验ꎮ实验结果表明:与YOLOv5s算法相比ꎬHWD ̄YOLOv5s算法在准确率、召回率、平均精度三个方面分别提升0.4%、1.1%、0.2%ꎻ检测速度能够达到实时检测要求ꎮ关键词:YOLOv5sꎻ目标检测ꎻ特征提取ꎻ特征融合中图分类号:TP391.41文献标志码:ADOI:10.3969/j.issn.1003-1251.2023.05.002AlgorithmofHelmetWearingDetectionBasedonImprovedYOLOv5sCHENYangꎬLYUYanhui(ShenyangLigongUniversityꎬShenyang110159ꎬChina)Abstract:Trafficregulationsstipulatethatelectricvehicledriversneedtowearsafetyhel ̄metswhendriving.Generaldetectionalgorithmshavetheproblemofmissingdetectionwhendetectingsmalltargetssuchassafetyhelmets.Inviewofthi...