Small object detection in remote sensing images is challenging. Traditional cNN downsampling often leads to the loss of small object details and missed detections. This paper proposes an improved YOLOv8 algorithm, inc...
详细信息
Small object detection in remote sensing images is challenging. Traditional cNN downsampling often leads to the loss of small object details and missed detections. This paper proposes an improved YOLOv8 algorithm, incorporating adaptive feature extraction and multi-scale fusion modules to enhance feature representation, effectively capturing the details of small objects. We introduced the AfGcAttention mechanism to strengthen the network's focus on key regions while suppressing irrelevant background information, improving the model's ability to recognize small objects. To address the resolution loss issue in small object detection, we adopted the cARAfE (content-Aware ReAssembly offEatures) upsampling operator. By performing content-aware reassembly offeature maps, cARAfE avoids the blurriness and information loss commonly associated with traditional upsampling methods, demonstrating significant advantages in reconstructing the details of small objects, resulting in clearer and more accurate boundaries. Additionally, to improve the accuracy of bounding box regression, we integrated the GIoU loss function to optimize the geometric matching between ground truth and predicted boxes, addressing the problem of inaccurate bounding box localization for small objects and enhancing localization precision. Experimental results demonstrate that the proposed algorithm significantly improves the precision of small object detection and maintains robustness in complex backgrounds. The improved model achieved an mAP of 83.0%, with accuracy improvements of 85.0%, 2.0%, and 5.0% compared to the baseline. compared with existing detection methods, this approach shows outstanding performance in detection accuracy, localization precision, and computational efficiency, particularly excelling in small object detection.
暂无评论