• LR-YOLO: 面向遥感图像的轻量级旋转目标检测算法

    LR-YOLO: Lightweight rotating object detection algorithm for remote sensing images

    • 针对遥感目标图像检测中存在的小目标检测困难、尺度变化大、方向任意以及在小型设备中检测精度不足等问题,提出了一种基于改进YOLOv8的轻量型遥感图像旋转目标检测算法LR-YOLO(Lightweight Rotating-YOLO)。首先,利用结构重参数技术并加入EMA注意力机制设计特征提取模块C2f-RA,增强网络的特征提取和特征表达能力,有效过滤复杂背景噪声的干扰。其次,在网络架构中采用ADown下采样,在减少特征图的尺寸的过程中防止上下文信息丢失过多。最后,针对检测头分类和定位两个分支缺乏交互,导致预测不一致的问题,提出任务动态对齐检测头ETDA-Head,促进分类和定位任务间的协同作用,以提高网络模型的精度。实验结果表明,改进后的算法在DIOR-R数据集和HRSC2016数据集上mAP@0.5上较基准模型分别提升1.4%和1.6%,参数量减少56.25%,相较于代表性的RoI Transformer算法检测精度提高了16.9%。这表明LR-YOLO算法有效地提升了检测精度,并显著降低模型的复杂度,验证了其有效性。

       

      Abstract: To address the challenges of detecting small objects in remote sensing images, such as difficulties with small object detection, large scale variations, arbitrary orientations, and insufficient detection accuracy on lightweight devices, this paper proposes a lightweight remote sensing image rotation object detection algorithm, LR-YOLO (Lightweight Rotating-YOLO), based on an improved YOLOv8 framework. First, the feature extraction module C2f-RA is designed using structural re-parameterization techniques and incorporating the EMA attention mechanism to enhance the network's feature extraction and representation capabilities, effectively filtering out complex background noise. Secondly, ADown sampling is employed in the network architecture to reduce feature map dimensions while minimizing the loss of contextual information. Finally, to address the issue of inconsistent predictions caused by the lack of interaction between the classification and localization branches of the detection head, a task-dynamically aligned detection head (ETDA-Head) is proposed. This promotes synergy between the classification and localization tasks, thereby improving the accuracy of the network model.Experimental results demonstrate that the improved algorithm achieves a 1.4% and 1.6% increase in detection accuracy on the DIOR-R and HRSC2016 datasets, respectively, compared to the baseline model, while reducing the number of parameters by 56.25%. Additionally, compared to the representative RoI Transformer algorithm, the detection accuracy is improved by 16.9%, indicating that the LR-YOLO algorithm significantly enhances detection accuracy while reducing model complexity, thereby validating its effectiveness.

       

    /

    返回文章
    返回