###
计算机系统应用英文版:2025,34(9):151-161
本文二维码信息
码上扫一扫!
改进YOLOv8的轻量化无人机航拍目标检测
(1.南京信息工程大学 江苏省大数据分析技术重点实验室, 南京 210044;2.江苏省大气环境与装备技术协同创新中心, 南京 210044)
Lightweight UAV Aerial Target Detection Based on Improved YOLOv8
(1.Jiangsu Key Laboratory of Big Data Analysis Technology, Nanjing University of Information Science & Technology, Nanjing 210044, China;2.Jiangsu Collaborative Innovation Center of Atmospheric Environment and Equipment Technology, Nanjing 210044, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 985次   下载 1292
Received:February 18, 2025    Revised:March 10, 2025
中文摘要: 无人机航拍场景下的目标检测因目标尺寸小、物体间遮挡严重、尺度变化大等因素, 常出现漏检与误检的问题. 此外, 受限于无人机平台的计算性能, 实现高精度与轻量化的实时目标检测具有较大挑战. 为此, 本文提出了一种改进YOLOv8的轻量级目标检测算法. 该算法采用轻量化的分割一切模型(segment anything model, SAM): MobileSAM的图像编码器作为YOLOv8的骨干网络, 能够有效地提取多尺度特征, 提升模型对小目标的检测效率, 同时提高泛化能力, 增强在不同任务和数据集上的表现. 针对检测头部分, 进行了轻量化设计, 提出基于共享卷积与自适应特征缩放的轻量化检测头LSCD (lightweight scalable shared convolutional detection head)以减少参数量和计算开销, 保持检测精度的同时降低模型参数量. 最后, 使用归一化高斯Wasserstein 距离(normalized Gaussian Wasserstein distance, NWD)损失, 提升小目标检测能力. 所提算法在VisDrone-DET2019数据集上对小目标的检测精度和召回率相较于原始YOLOv8s模型有较大提升; 相比于原始YOLOv8s模型, mAP50提高了3.2%, 达到41.4%, 且参数量减少了33.9%. 在DOTA v1.0数据集上, mAP50达到48.8%, 提升了8%, 表明算法具有较好的泛化能力.
Abstract:Target detection from a UAV perspective often faces challenges such as small object size, severe occlusion, and significant scale variation, which result in missed and false detections. Moreover, constrained by the limited computational capacity of UAV platforms, achieving real-time detection with both high accuracy and lightweight deployment remains difficult. To address these challenges, a lightweight version of YOLOv8 is proposed. The image encoder of MobileSAM, a lightweight adaptation of the segment anything model (SAM), is integrated as the backbone network, enabling effective extraction of multi-scale features, improved detection efficiency for small targets, and enhanced generalization across diverse tasks and datasets. A lightweight detection head, termed lightweight scalable shared convolutional detection head (LSCD), is proposed. Based on shared convolution and adaptive feature scaling, LSCD reduces the parameter count and computational overhead while maintaining detection accuracy. In addition, the normalized Gaussian Wasserstein distance (NWD) loss is employed to further improve detection performance on small objects. Experimental results on the VisDrone-DET2019 dataset demonstrate that the proposed algorithm significantly enhances both precision and recall for small targets compared to the original YOLOv8s model. The mAP50 is increased by 3.2%, reaching 41.4%, while the number of parameters is reduced by 33.9%. On the DOTA v1.0 dataset, the proposed model achieves a mAP50 of 48.8%, representing an 8% improvement, indicating strong generalization capability.
文章编号:     中图分类号:    文献标志码:
基金项目:国家自然科学基金(61872189)
引用文本:
严嘉旭,苏天康,宋慧慧.改进YOLOv8的轻量化无人机航拍目标检测.计算机系统应用,2025,34(9):151-161
YAN Jia-Xu,SU Tian-Kang,SONG Hui-Hui.Lightweight UAV Aerial Target Detection Based on Improved YOLOv8.COMPUTER SYSTEMS APPLICATIONS,2025,34(9):151-161