Deploying target detection models on edge devices such as UAVs is challenging due to their limited size and computational capacity, while target detection models typically require significant computational resources. To address this issue, this study proposes a lightweight real-time infrared object detection model named LRI-YOLO (Lightweight Real-time Infrared YOLO), which is based on YOLOv8n. The model improves the C2f module’s Bottleneck structure by integrating Partial Convolution (PConv) with Pointwise Convolution (PWConv), achieving a more lightweight design. Furthermore, during the feature fusion stage, the original downsampling structure with ordinary convolution is replaced with a combination of max pooling and regular convolution. This modification retains more feature map information. The model’s structure is further optimized by redesigning the decoupled detection head with Group Convolution (GConv) instead of ordinary convolution, significantly enhancing detection speed. Additionally, the original BCELoss is replaced with EMASlideLoss, a newly developed classification loss function introduced in this study. This loss function allows the model to focus more on hard samples, thereby improving its classification capability. Compared to the YOLOv8n algorithm, LRI-YOLO is more lightweight, with its parameters reduced by 46.7% and floating-point operations (FLOPs) reduced by 53.1%. Moreover, the mean average precision (mAP) reached 94.1%. Notably, on devices with moderate computational power that only have a Central Processing Unit (CPU), the detection speed reached 42 frames per second (FPS), surpassing most mainstream models. This indicates that LRI-YOLO offers a novel solution for real-time infrared object detection on edge devices such as drones.


    Access

    Download


    Export, share and cite



    Title :

    A Lightweight Real-Time Infrared Object Detection Model Based on YOLOv8 for Unmanned Aerial Vehicles


    Contributors:
    Baolong Ding (author) / Yihong Zhang (author) / Shuai Ma (author)


    Publication date :

    2024




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    Unknown




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

    Xiaofeng Zhao / Wenwen Zhang / Hui Zhang et al. | DOAJ | 2024

    Free access


    Monitoring Jackfruit Growth Based on YOLOv8 with Unmanned Aerial Vehicles

    Han, Haoyuan / Wang, Zijun / Wu, Di | IEEE | 2024


    An Unmanned Aerial Vehicle Image Object Detection Algorithm Based on Improved YOLOv8

    Ren, Xilin / Zhang, Zegu / Yang, Fanghua et al. | IEEE | 2024


    UAV Detection for Aerial Vehicles using YOLOv8

    Sairam, D.V. / Bhuvaneswari, R / Chokkalingam, SP | IEEE | 2024