Laser Journal, Volume. 45, Issue 8, 61(2024)

A lightweight giant panda object detection model with attention mechanism

LYU Haotian and JIA Xiaolin*
Author Affiliations
  • Mobile Internet of Things and Radio Frequency Identification Technology Key Laboratory of Mianyang, School of Computer Science and Technology, Southwest University of Science and Technology, Mianyang Sichuan 621010, China
  • show less

    To address the problem of poor detection performance of giant pandas in complex environments and the low efficiency of object detection models on resource-limited embedded devices, a lightweight giant panda detection model called GP-YOLOv5n is proposed. The model is based on YOLOv5n and improved by introducing a depth-separable neck network with attention, which enhances the detection accuracy and speed of targets in complex environments. Moreover, the model adopts Alpha-IoU in the bounding box regression loss function to improve the bounding box localization accuracy of targets. After training on a homemade giant panda dataset, the model is optimized for embedded devices and deployed on Jetson Nano. Experimental results show that the improved model achieves 97.8% and 73.6% in mAP50 and mAP50:95 metrics respectively, which are 2.7% and 9.2% higher than the original model. The detection speed of the model on embedded devices reaches 15.12 f/s, which can accurately and real-time detect the giant pandas in complex environments.

    Tools

    Get Citation

    Copy Citation Text

    LYU Haotian, JIA Xiaolin. A lightweight giant panda object detection model with attention mechanism[J]. Laser Journal, 2024, 45(8): 61

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Jun. 7, 2023

    Accepted: Dec. 20, 2024

    Published Online: Dec. 20, 2024

    The Author Email: Xiaolin JIA (my_jiaxl@163.com)

    DOI:10.14016/j.cnki.jgzz.2024.08.061

    Topics