文档名：基于改进YOLOv7的小目标交通标志检测算法
摘要：在自动驾驶中,检测小目标的精度和漏检问题对于车辆设备对路况判断的准确性至关重要.针对小交通标志检测精度漏检的问题,文中提出一种基于YOLOv7的小目标交通标志检测算法.在Neck中引入ACmix注意力模块,利用ACmix注意力模块提高网络对小尺度目标的敏感度,降低噪声所带来的影响;在输出端引入多头自注意力机制SPD卷积构建块,提升输出端对交通标志小目标的检测性能;使用SIoU替换原始YOLOv7网络中的CIoU来优化损失函数,提高网络鲁棒性.为验证所提算法的性能,在CCTSDB交通标志数据集上进行了实验验证.实验结果表明,对于数据集中的小目标,改进后的YOLOv7网络相比于原始的YOLOv7网络,漏检情况得到了明显改善,且map达到了98.7％,运行速度达到了110.5fps.相较于原始YOLOv7模型,精度和速度均有提升,可以满足交通标志检测的要求.
Abstract：Inautomaticdriving,theproblemofdetectingtheaccuracyandleakageofsmalltargetsiscru-cialfortheaccuracyofvehicleequipmentinjudgingroadconditions.Aimingattheproblemofsmalltraf-ficsigndetectionaccuracyandmisseddetection,asmalltargettrafficsigndetectionalgorithmbasedonimprovedYOLOv7isproposed.TheACmixattentionmoduleisintroducedintoNeck,whichisusedtoimprovethesensitivityofthenetworktosmall-scaletargetsandreducetheimpactofnoise;themulti-headself-attentionmechanismSPDconvolutionalbuildingblockisintroducedattheoutputsidetoim-provethedetectionperformanceofsmalltrafficsigntargetsattheoutputside;andthelossfunctionisop-timizedbyusingSIoUtoreplacetheCIoUintheoriginalYOLOv7networkthatimprovestherobustnessofthenetwork.Toverifytheperformanceoftheproposedalgorithm,experimentalvalidationiscarriedoutontheCCTSDBtrafficsigndataset.Theexperimentalresultsshowthatforthesmalltargetsinthedata-set,theimprovedYOLOv7network,comparedwiththeoriginalYOLOv7network,theleakagedetectionissignificantlyimprovedandthemapreaches98.7％,andtheoperationspeedreaches110.5f/s,whichisanimprovementinbothaccuracyandspeedcomparedwiththeoriginalYOLOv7model,anditcansat-isfytherequirementsoftrafficsigndetection.
作者：郑娟毅 杨溥江 郭梦月 董嘉豪 张庆珏Author：ZHENGJuan-yi YANGPu-jiang GUOMeng-yue DONGJia-hao ZHANGQing-jue
作者单位：西安邮电大学通信与信息工程学院,陕西西安710121
刊名：中国电子科学研究院学报
Journal：JournalofChinaAcademyofElectronicsandInformationTechnology
年，卷(期)：2024, 19(3)
分类号：TP391.41
关键词：交通标志检测 YOLOv7 小目标 SPD SIoU ACmix
Keywords：trafficsigndetection YOLOv7 smalltarget SPD SIoU ACmix
机标分类号：TP391U471.15O225
在线出版日期：2024年7月2日
基金项目：国家自然科学基金基于改进YOLOv7的小目标交通标志检测算法[
期刊论文] 中国电子科学研究院学报--2024, 19(3)郑娟毅 杨溥江 郭梦月 董嘉豪 张庆珏在自动驾驶中,检测小目标的精度和漏检问题对于车辆设备对路况判断的准确性至关重要.针对小交通标志检测精度漏检的问题,文中提出一种基于YOLOv7的小目标交通标志检测算法.在Neck中引入ACmix注意力模块,利用ACmix注意力...参考文献和引证文献
参考文献
引证文献
本文读者也读过
相似文献
相关博文
基于改进YOLOv7的小目标交通标志检测算法 Small Target Traffic Sign Detection Algorithm Based on Improved YOLOv7
基于改进YOLOv7的小目标交通标志检测算法.pdf
- 文件大小:
- 14.94 MB
- 下载次数:
- 60
-
高速下载
|
|
