文档名：基于YOLOv5的轻量化无人机航拍小目标检测算法
摘要：针对无人机航拍视角下图像目标特征尺寸小且存在背景复杂、分布密集的问题,提出了一种基于YOLOv5的轻量化无人机航拍小目标检测改进算法GA-YOLO.该算法改进了Mosaic数据增强方法和网络整体结构,并增加了微小物体检测头,同时设计了轻量化的全局注意力模块和并行结构的空间通道注意力机制模块,提高了网络的全局特征提取能力和训练过程中卷积通道之间的竞争和合作关系.以4.0版本的YOLOv5s为基准,在公开无人机航拍数据集VisDrone2019-DET上实验,结果表明,改进后的模型相较于原模型,参数量下降了48%,计算量下降了26%,而mAP@0.5提高了4.9个百分点,mAP@0.5:0.95提高了3.3个百分点,有效地提高了无人机空中视角下对密集型小目标的检测能力.
Abstract：Alightweightunmannedaerialvehicle(UAV)aerialphotographysmalltargetdetectionalgorithmGA-YOLObasedonYOLOv5isproposedtoaddresstheproblemofsmalltargetfeaturesize,complexbackground,anddensedistributioninimagesfromtheperspectiveofUAVaerialphotography.ThisalgorithmimprovestheMosaicdataaugmentationmethodandoverallnetworkstructure,andaddsasmallobjectdetectionhead.Atthesametime,alightweightglobalattentionmoduleandaparallelspatialchannelattentionmechanismmodulearedesignedtoenhancethenetwork'sglobalfeatureextractionabilityandthecompetitionandcooperationbetweenconvolutionalchannelsduringthetrainingprocess.Basedonthe4.0versionofYOLOv5s,experimentswereconductedonthepubliclyavailabledroneaerialphotographydatasetVisDrone2019-DET.Theresultsshowedthattheimprovedmodelreducedthenumberofparametersby48%andthecomputationalcomplexityby26%comparedtotheoriginalmodel,andmAP@0.5improvedby4.9percentagepoints,mAP@0.50.95increasedby3.3percentagepoints,effectivelyenhancingthedetectioncapabilityofunmannedaerialvehiclesfordensesmalltargetsfromanaerialperspective.
作者：李雪森 谭北海 余荣 薛先斌 Author：LiXue-sen TanBei-hai YuRong XueXian-bin
作者单位：广东工业大学自动化学院,广东广州510006广东工业大学集成电路学院,广东广州510006
刊名：广东工业大学学报
Journal：JournalofGuangdongUniversityofTechnology
年，卷(期)：2024, 41(3)
分类号：TP391.41
关键词：无人机航拍 YOLOv5s 小目标检测 数据增强 注意力机制
Keywords：UAVaerialphotography YOLOv5s smalltargetdetection dataenhancement attentionmechanism
机标分类号：TP391.41TN911.7TP277
在线出版日期：2024年6月19日
基金项目：国家自然科学基金，国家自然科学基金，广东省基础与应用基础研究基金联合基金重点项目，广西自然科学基金重点项目基于YOLOv5的轻量化无人机航拍小目标检测算法[
期刊论文] 广东工业大学学报--2024, 41(3)李雪森 谭北海 余荣 薛先斌针对无人机航拍视角下图像目标特征尺寸小且存在背景复杂、分布密集的问题,提出了一种基于YOLOv5的轻量化无人机航拍小目标检测改进算法GA-YOLO.该算法改进了Mosaic数据增强方法和网络整体结构,并增加了微小物体检测头,同...参考文献和引证文献
参考文献
引证文献
本文读者也读过
相似文献
相关博文
基于YOLOv5的轻量化无人机航拍小目标检测算法 Small Target Detection Algorithm for Lightweight UAV Aerial Photography Based on YOLOv5
基于YOLOv5的轻量化无人机航拍小目标检测算法.pdf
- 文件大小:
- 2.12 MB
- 下载次数:
- 60
-
高速下载
|
|
