Abstract:A lightweight infrared target detection model, G-YOLO, based on an unmanned aerial vehicle (UAV) is proposed to address the issues of low accuracy in target detection of UAV aerial images in complex ground scenarios and large network models that are difficult to apply to mobile or embedded platforms. Firstly, the YOLOv8 backbone feature extraction network is improved and designed based on the lightweight network, GhostBottleneckV2, and the remaining part of the backbone network adopts the depth-separable convolution, DWConv, to replace part of the standard convolution, which effectively retains the detection effect of the model while greatly reducing the number of model parameters and calculations. Secondly, the neck structure is improved by the ODConv module, which adopts an adaptive convolutional structure to adaptively adjust the convolutional kernel size and step size, which allows for more effective feature extraction and detection based on targets at different scales. At the same time, the neck structure is further optimized using the attention mechanism, SEAttention, to improve the model’s ability to learn global information of input feature maps, which is then applied to each channel of each feature map to enhance the useful information in a specific channel and improve the model’s detection performance. Finally, the introduction of the SlideLoss loss function enables the model to calculate the differences between predicted and actual truth bounding boxes during the training process, and adjust the model parameters based on these differences to improve the accuracy and efficiency of object detection. The experimental results show that compared with YOLOv8n, the G-YOLO reduces the missed and false detection rates of infrared small target detection in complex backgrounds. The number of model parameters is reduced by 74.2%, the number of computational floats is reduced by 54.3%, the FPS is improved by 71, which improves the detection efficiency of the model, and the average accuracy (mAP) reaches 91.4%, which verifies the validity of the model for UAV-based infrared small target detection. Furthermore, the FPS of the model reaches 556, and it will be suitable for wider and more complex detection task such as small targets, long-distance targets, and other complex scenes.

Object Detection for Rescue Operations by High-Altitude Infrared Thermal Imaging Collected by Unmanned Aerial Vehicles

Object Detection from UAV Thermal Infrared Images and Videos Using YOLO Models.

Research on Aerial Photography Target Detection Methods

ITD-YOLOv8: an Infrared Target Detection Model Based on YOLOv8 for Unmanned Aerial Vehicles

UAV Infrared Image Human and Vehicle Object Detection Based on YOLOv5

LightUAV-YOLO: a Lightweight Object Detection Model for Unmanned Aerial Vehicle Image

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

Analysis and Adaptation of YOLOv4 for Object Detection in Aerial Images

Real Time Human Detection by Unmanned Aerial Vehicles

Performance Comparison of YOLOv5 and YOLOv8 Architectures in Human Detection using Aerial Images

Infra-YOLO: Efficient Neural Network Structure with Model Compression for Real-Time Infrared Small Object Detection

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

Target Detection Method of UAV Aerial Imagery Based on Improved YOLOv5

Small object detection based on YOLOv8 in UAV perspective

RMT-YOLOv9s: An Infrared Small Target Detection Method Based on UAV Remote Sensing Images

Object Detection Deployed on UAVs for Oblique Images by Fusing IMU Information

YOLO-ViT-Based Method for Unmanned Aerial Vehicle Infrared Vehicle Target Detection

YOLOSR-IST: A Deep Learning Method for Small Target Detection in Infrared Remote Sensing Images based on Super-Resolution and YOLO

G-YOLO: A Lightweight Infrared Aerial Remote Sensing Target Detection Model for UAVs Based on YOLOv8

Lightweight Low-Altitude UAV Object Detection Based on Improved YOLOv5s

YOLO-ERF: Lightweight Object Detector for UAV Aerial Images