IDEAS home Printed from https://ideas.repec.org/a/gam/jagris/v14y2024i7p1052-d1425901.html
   My bibliography  Save this article

A Lightweight Rice Pest Detection Algorithm Using Improved Attention Mechanism and YOLOv8

Author

Listed:
  • Jianjun Yin

    (College of Mathematics and Informatics, South China Agricultural University, Guangzhou 510642, China
    Key Laboratory of Smart Agricultural Technology in Tropical South China, Ministry of Agriculture and Rural Affairs, Guangzhou 510642, China)

  • Pengfei Huang

    (College of Mathematics and Informatics, South China Agricultural University, Guangzhou 510642, China
    Key Laboratory of Smart Agricultural Technology in Tropical South China, Ministry of Agriculture and Rural Affairs, Guangzhou 510642, China)

  • Deqin Xiao

    (College of Mathematics and Informatics, South China Agricultural University, Guangzhou 510642, China
    Key Laboratory of Smart Agricultural Technology in Tropical South China, Ministry of Agriculture and Rural Affairs, Guangzhou 510642, China)

  • Bin Zhang

    (College of Mathematics and Informatics, South China Agricultural University, Guangzhou 510642, China
    Key Laboratory of Smart Agricultural Technology in Tropical South China, Ministry of Agriculture and Rural Affairs, Guangzhou 510642, China)

Abstract

Intelligent pest detection algorithms are capable of effectively detecting and recognizing agricultural pests, providing important recommendations for field pest control. However, existing recognition models have shortcomings such as poor accuracy or a large number of parameters. Therefore, this study proposes a lightweight and accurate rice pest detection algorithm based on improved YOLOv8. Firstly, a Multi-branch Convolutional Block Attention Module (M-CBAM) is constructed in the YOLOv8 network to enhance the feature extraction capability for pest targets, yielding better detection results. Secondly, the Minimum Points Distance Intersection over Union (MPDIoU) is introduced as a bounding box loss metric, enabling faster model convergence and improved detection results. Lastly, lightweight Ghost convolutional modules are utilized to significantly reduce model parameters while maintaining optimal detection performance. The experimental results demonstrate that the proposed method outperforms other detection models, with improvements observed in all evaluation metrics compared to the baseline model. On the test set, this method achieves a detection average precision of 95.8% and an F1-score of 94.6%, with a model parameter of 2.15 M, meeting the requirements of both accuracy and lightweightness. The efficacy of this approach is validated by the experimental findings, which provide specific solutions and technical references for intelligent pest detection.

Suggested Citation

  • Jianjun Yin & Pengfei Huang & Deqin Xiao & Bin Zhang, 2024. "A Lightweight Rice Pest Detection Algorithm Using Improved Attention Mechanism and YOLOv8," Agriculture, MDPI, vol. 14(7), pages 1-17, June.
  • Handle: RePEc:gam:jagris:v:14:y:2024:i:7:p:1052-:d:1425901
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2077-0472/14/7/1052/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2077-0472/14/7/1052/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Qixun Xiao & Wenying Zheng & Yifan He & Zijie Chen & Fanxin Meng & Liyan Wu, 2023. "Research on the Agricultural Pest Identification Mechanism Based on an Intelligent Algorithm," Agriculture, MDPI, vol. 13(10), pages 1-17, September.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.

      Corrections

      All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jagris:v:14:y:2024:i:7:p:1052-:d:1425901. See general information about how to correct material in RePEc.

      If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

      If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

      If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

      For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

      Please note that corrections may take a couple of weeks to filter through the various RePEc services.

      IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.