首页|期刊导航|哈尔滨商业大学学报(自然科学版)|基于改进YOLOv8s的航拍小目标检测算法

基于改进YOLOv8s的航拍小目标检测算法OA

Aerial small object detection algorithm based on improved YOLOv8s

中文摘要英文摘要

针对无人机航拍小目标检测中存在检测精度低、误检、漏检以及模型参数量大等难题,提出一种改进YOLOv8s的航拍小目标检测算法.引入RepViTBlock轻量化模块改进骨干网络和颈部网络中的C2f模块,并引入EMA注意力机制进一步改进骨干网络中的C2f模块,提升特征提取能力并降低了模型的参数量.使用三重尺度序列编码融合模块TSEF对颈部网络进行重构,并融合构建了小目标检测层P2,在降低了参数量的同时提升了检测精度.最后利用Inner-CIoU改进损失函数,提升模型边框回归的性能和检测精度.实验结果表明,在VisDrone2019航拍数据集上,改进算法的精准率P、召回率R、平均检测精度mAP50分别为54.2%、42.3%、43.7%,相较于YOLOv8s分别提升了5.7%、8.5%、11.5%,参数量降低了38.7%,适用于无人机目标检测任务.

Aiming at the problems such as low detection accuracy,false detection,missed detection,and large number of model parameters in the detection of small targets in unmanned aerial vehicle(UAV)aerial photography,an improved YOLOv8s aerial photography small target detection algorithm was proposed.The RepViTBlock lightweight modulewas introducedto improve the C2f module in the backbone network and the neck network,and the EMA attention mechanism was introduced to further improve the C2f module in the backbone network,enhancing the feature extraction ability and reducing the number of parameters of the model.The tripartite scale sequence coding fusion module TSEF was used to reconstruct the neck network,and the small target detection layer P2 was fused and constructed,improving the detection accuracy while reducing the number of parameters.The loss function was improved by using Inner-CIoU to enhance the performance and detection accuracy of the border regression of the model.The experimental results showed that on the VisDrone 2019 aerial photography dataset,the precision P,recall R,and average detection accuracy mAP50 of the improved algorithm were 54.2%,42.3%,and 43.7%respectively.Compared with YOLOv8s,they increased by 5.7%,8.5%,and 11.5%respectively,and the number of parameters decreased by 38.7%.The improved algorithm was applicable to unmanned aerial vehicle(UAV)target detection tasks.

章子龙;王冠凌;熊伟;王坤相;王世康

安徽工程大学电气工程学院,安徽芜湖 241000安徽工程大学电气工程学院,安徽芜湖 241000安徽工程大学电气工程学院,安徽芜湖 241000安徽工程大学电气工程学院,安徽芜湖 241000安徽工程大学电气工程学院,安徽芜湖 241000

航空航天

小目标检测YOLOv8sRepViTBlockEMA注意力机制Inner-CIoU

small target detectionYOLOv8sRepViTBlockEMA attention mechanismInner-CIoU

《哈尔滨商业大学学报(自然科学版)》 2026 (1)

23-33,11

国家自然科学基金项目(62005293)

评论