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

被引:0
|
作者
Yin, Jianjun [1 ,2 ]
Huang, Pengfei [1 ,2 ]
Xiao, Deqin [1 ,2 ]
Zhang, Bin [1 ,2 ]
机构
[1] South China Agr Univ, Coll Math & Informat, Guangzhou 510642, Peoples R China
[2] Minist Agr & Rural Affairs, Key Lab Smart Agr Technol Trop South China, Guangzhou 510642, Peoples R China
来源
AGRICULTURE-BASEL | 2024年 / 14卷 / 07期
关键词
pest detection; YOLOv8; attention mechanism; loss metric; lightweight model;
D O I
10.3390/agriculture14071052
中图分类号
S3 [农学(农艺学)];
学科分类号
0901 ;
摘要
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.
引用
收藏
页数:17
相关论文
共 50 条
  • [41] An Improved YOLOv8 Algorithm for Rail Surface Defect Detection
    Wang, Yan
    Zhang, Kehua
    Wang, Ling
    Wu, Lintong
    IEEE ACCESS, 2024, 12 : 44984 - 44997
  • [42] Research on improved YOLOv8 algorithm for insulator defect detection
    Zhang, Lin
    Li, Boqun
    Cui, Yang
    Lai, Yushan
    Gao, Jing
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2024, 21 (01)
  • [43] UAV Target Detection Algorithm Based on Improved YOLOv8
    Wang, Feng
    Wang, Hongyuan
    Qin, Zhiyong
    Tang, Jiaying
    IEEE ACCESS, 2023, 11 : 116534 - 116544
  • [44] Improved Road Defect Detection Algorithm Based on YOLOv8
    Wang, Xueqiu
    Gao, Huanbing
    Jia, Zemeng
    Computer Engineering and Applications, 2024, 60 (17) : 179 - 190
  • [45] An Improved Liver Disease Detection Based on YOLOv8 Algorithm
    Huang, Junjie
    Li, Caihong
    Yan, Fengjun
    Guo, Yuanchun
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2024, 15 (07) : 1168 - 1179
  • [46] Improved YOLOv8 Algorithm for Water Surface Object Detection
    Wang, Jie
    Zhao, Hong
    SENSORS, 2024, 24 (15)
  • [47] POTATO APPEARANCE DETECTION ALGORITHM BASED ON IMPROVED YOLOv8
    Zhang, Huan
    Liu, Zhen
    Yang, Ranbing
    Pan, Zhiguo
    Su, Zhaoming
    Li, Xinlin
    Liu, Zeyang
    Shi, Chuanmiao
    Wang, Shuai
    Wu, Hongzhu
    INMATEH - Agricultural Engineering, 2024, 74 (03): : 864 - 874
  • [48] Pest Detection in Olive Groves Using YOLOv7 and YOLOv8 Models
    Alves, Adilia
    Pereira, Jose
    Khanal, Salik
    Morais, A. Jorge
    Filipe, Vitor
    OPTIMIZATION, LEARNING ALGORITHMS AND APPLICATIONS, PT II, OL2A 2023, 2024, 1982 : 50 - 62
  • [49] A lightweight Yunnan Xiaomila detection and pose estimation based on improved YOLOv8
    Wang, Fenghua
    Tang, Yuan
    Gong, Zaipeng
    Jiang, Jin
    Chen, Yu
    Xu, Qiang
    Hu, Peng
    Zhu, Hailong
    FRONTIERS IN PLANT SCIENCE, 2024, 15
  • [50] A lightweight model for echo trace detection in echograms based on improved YOLOv8
    Jungang Ma
    Jianfeng Tong
    Minghua Xue
    Junfan Yao
    Scientific Reports, 14 (1)