本文已被:浏览 985次 下载 1292次
Received:February 18, 2025 Revised:March 10, 2025
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.
keywords: UAV aerial perspective target detection YOLOv8 lightweight normalized Gaussian Wasserstein distance (NWD) MobileSAM lightweight scalable shared convolutional detection head (LSCD)
文章编号: 中图分类号: 文献标志码:
基金项目:国家自然科学基金(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
严嘉旭,苏天康,宋慧慧.改进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

