改进YOLOv8检测航拍图像中小目标,兼顾速度与精度。
Efficient Oriented Object Detection with Enhanced Small Object Recognition in Aerial Images
- 引入小波变换和自适应特征金字塔提升小目标感知能力。
- 参数量2160万,比DecoupleNet少170万,推理更快。
- 适合算力受限的无人机或边缘设备部署使用。
在航拍图像的旋转边界框目标检测中,实现计算效率与检测精度的平衡仍具挑战。现有轻量级模型在遥感影像中小目标和多尺度目标检测上表现不足。为此,我们对YOLOv8进行改进,提出面向旋转目标检测的高效架构:基于小波变换的C2f模块用于捕捉关联特征,自适应尺度特征金字塔(ASFP)利用P2层细节信息;引入GhostDynamicConv进一步降低模型复杂度。该模型参数量为2160万,低于DecoupleNet的2330万,在DOTAv1.0数据集上达到与领先方法相当的平均精度(mAP)。其高效设计使其在资源受限环境下尤为适用。
原文摘要 · Abstract (English)
Achieving a balance between computational efficiency and detection accuracy in the realm of rotated bounding box object detection within aerial imagery is a significant challenge. While prior research has aimed at creating lightweight models that enhance computational performance and feature extraction, there remains a gap in the performance of these networks when it comes to the detection of small and multi-scale objects in remote sensing (RS) imagery. To address these challenges, we present a novel enhancement to the YOLOv8 model, tailored for oriented object detection tasks and optimized for environments with limited computational resources. Our model features a wavelet transform-based C2f module for capturing associative features and an Adaptive Scale Feature Pyramid (ASFP) module that leverages P2 layer details. Additionally, the incorporation of GhostDynamicConv significantly contributes to the model's lightweight nature, ensuring high efficiency in aerial imagery analysis. Featuring a parameter count of 21.6M, our approach provides a more efficient architectural design than DecoupleNet, which has 23.3M parameters, all while maintaining detection accuracy. On the DOTAv1.0 dataset, our model demonstrates a mean Average Precision (mAP) that is competitive with leading methods such as DecoupleNet. The model's efficiency, combined with its reduced parameter count, makes it a strong candidate for aerial object detection, particularly in resource-constrained environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。