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.

YOLOv7-Aerial: Rotated Object Detection of Aerial Images Based on Improved YOLOv7

Rotated YOLOv4 with Attention-wise Object Detectors in Aerial Images

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

Efficient-Lightweight YOLO: Improving Small Object Detection in YOLO for Aerial Images

Efficient YOLOv7-Drone: An Enhanced Object Detection Approach for Drone Aerial Imagery

Aerial image object detection based on improved YOLOv5

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

YOLOv7-P: a lighter and more effective UAV aerial photography object detection algorithm

Lightweight and Efficient Tiny-Object Detection Based on Improved YOLOv8n for UAV Aerial Images

Research on a small target object detection method for aerial photography based on improved YOLOv7

Lightweight unmanned aerial vehicle object detection algorithm based on improved YOLOv8

An Aerial Image Detection Algorithm Based on Improved YOLOv5

A Small Object Detection Method for Drone-Captured Images Based on Improved YOLOv7

Efficient Small Object Detection You Only Look Once: A Small Object Detection Algorithm for Aerial Images

A Modified YOLOv8 Detection Network for UAV Aerial Image Recognition

From Blurry to Brilliant Detection: YOLOv5-Based Aerial Object Detection with Super Resolution

YOLOD: A Target Detection Method for UAV Aerial Imagery

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

R-YOLO: A YOLO-Based Method for Arbitrary-Oriented Target Detection in High-Resolution Remote Sensing Images

YOLO-Drone: An Optimized YOLOv8 Network for Tiny UAV Object Detection