Abstract:Aerial image object detection has a wide range of application values in civilian or military fields. Due to its unique high-altitude imaging viewpoint and the multiangle shooting method, aerial images lead to problems, such as small objects being detected in the image, large variations in object scales, and dense distribution. To alleviate the above problems, we propose an improved aerial image object detection algorithm aerial images object detection based on YOLO (AIOD-YOLO) based on YOLOv8-s. First, we propose the multibranch contextual information aggregation module. It enhances the network’s perception of small objects by associating object information with the surrounding environment, thereby compensating for the lack of feature information for small objects. In addition, we propose the multilayer feature cascade efficient aggregation network, which leverages multigradient flow fusion of features at different scales. This approach aids the network in capturing a wide range of scale information and effectively mitigates the issue of missed detections caused by variations in object scales. Finally, we propose the adaptive task alignment label assignment strategy to address the issue of dense object distribution. The strategy incorporates the cosine similarity calculation to assess alignment globally and simultaneously adjusts the weights of positive and negative samples. We optimize the precision of label assignment for dense objects in aerial images, effectively resolving the challenges posed by their close proximity. The experiments on the VisDrone dataset demonstrate that AIOD-YOLO achieves a significant 7.2% improvement in mAP compared to the baseline model YOLOv8-s. The mAP0.5 of AIOD-YOLO is also improved by 14.1%, 7.9%, and 7.5% on SeaDronesSee v2, AI-TOD, and TinyPerson datasets, respectively, which validates the generalization of our proposed algorithm. AIOD-YOLO offers a superior information processing approach for tasks related to aerial image object detection in both civilian and military applications.

Multi-Object Detection Algorithm for UAV Aerial Images based on Improved YOLOV8

Lightweight unmanned aerial vehicle object detection algorithm based on improved YOLOv8

UAV-YOLOv8: A Small-Object-Detection Model Based on Improved YOLOv8 for UAV Aerial Photography Scenarios

ARF-YOLOv8: a novel real-time object detection model for UAV-captured images detection

YOLOv7-UAV: An Unmanned Aerial Vehicle Image Object Detection Algorithm Based on Improved YOLOv7

Small Object Detection in UAV Images Based on YOLOv8n

An Efficient UAV Image Object Detection Algorithm Based on Global Attention and Multi-Scale Feature Fusion

UAV Target Detection Algorithm Based on Improved YOLOv8

Target Detection Method of UAV Aerial Imagery Based on Improved YOLOv5

AIOD-YOLO: an algorithm for object detection in low-altitude aerial images

A Small-Object Detection Model Based on Improved YOLOv8s for UAV Image Scenarios

A UAV Aerial Image Target Detection Algorithm Based on YOLOv7 Improved Model

SMFF-YOLO: A Scale-Adaptive YOLO Algorithm with Multi-Level Feature Fusion for Object Detection in UAV Scenes

SOD-YOLO: Small-Object-Detection Algorithm Based on Improved YOLOv8 for UAV Images

A novel small object detection algorithm for UAVs based on YOLOv5

Small Object Detection in UAV Remote Sensing Images Based on Intra-Group Multi-Scale Fusion Attention and Adaptive Weighted Feature Fusion Mechanism

URS-YOLOv5s: Object Detection Algorithm for UAV Remote Sensing Images

Research on Object Detection and Recognition Method for UAV Aerial Images Based on Improved YOLOv5

An Improved YOLOv5 Method for Small Object Detection in UAV Capture Scenes

A Modified YOLOv5 for Object Detection in UAV-captured Scenarios