Abstract:A UAV infrared target detection model ITD-YOLOv8 based on YOLOv8 is proposed to address the issues of model missed and false detections caused by complex ground background and uneven target scale in UAV aerial infrared image target detection, as well as high computational complexity. Firstly, an improved YOLOv8 backbone feature extraction network is designed based on the lightweight network GhostHGNetV2. It can effectively capture target feature information at different scales, improving target detection accuracy in complex environments while remaining lightweight. Secondly, the VoVGSCSP improves model perceptual abilities by referencing global contextual information and multiscale features to enhance neck structure. At the same time, a lightweight convolutional operation called AXConv is introduced to replace the regular convolutional module. Replacing traditional fixed-size convolution kernels with convolution kernels of different sizes effectively reduces the complexity of the model. Then, to further optimize the model and reduce missed and false detections during object detection, the CoordAtt attention mechanism is introduced in the neck of the model to weight the channel dimensions of the feature map, allowing the network to pay more attention to the important feature information, thereby improving the accuracy and robustness of object detection. Finally, the implementation of XIoU as a loss function for boundary boxes enhances the precision of target localization. The experimental findings demonstrate that ITD-YOLOv8, in comparison to YOLOv8n, effectively reduces the rate of missed and false detections for detecting multi-scale small targets in complex backgrounds. Additionally, it achieves a 41.9% reduction in model parameters and a 25.9% decrease in floating-point operations. Moreover, the mean accuracy (mAP) attains an impressive 93.5%, thereby confirming the model’s applicability for infrared target detection on unmanned aerial vehicles (UAVs).

Infrared Long-distance Target Detection Based on Deep Learning

Research on Infrared Image Target Detection Technology Based on YOLOv3 and Computer Vision

Yolo-Based Detection Technology For Aerial Infrared Targets

Research on Infrared Dim Target Detection Based on Improved YOLOv8

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

A YOLO Network Based on Depthwise Convolution Attention, Feature Fusion, and KL Divergence (DFK-YOLO): A Deep Learning Method for Infrared Small Target Detection Based on YOLOv7

YOLO-MST: Multiscale deep learning method for infrared small target detection based on super-resolution and YOLO

Infrared target detection using deep learning algorithms

Dual-YOLO Architecture from Infrared and Visible Images for Object Detection

Infrared Dim and Small Target Detection Based on Local–Global Feature Fusion

Research on Infrared Image Enhancement and Target Detection Algorithm Based on Deep Learning

Design Compact YOLO based Network for Small Target Detection on Infrared Image

LFIR-YOLO: Lightweight Model for Infrared Vehicle and Pedestrian Detection

An infrared small target detection model via Gather-Excite attention and normalized Wasserstein distance

YOLO-CIR: The network based on YOLO and ConvNeXt for infrared object detection

Infrared target detection and recognition in complex scene

YOLO-FIRI: Improved YOLOv5 for Infrared Image Object Detection

Lightweight Design for Infrared Dim and Small Target Detection in Complex Environments

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

A Novel Infrared Small Target Detection Algorithm Based on Deep Learning

Multiscale and Direction Target Detecting in Remote Sensing Images via Modified YOLO-v4