Rice Disease Detection Based on Improved YOLOv8
Author:
  • Article
  • | |
  • Metrics
  • |
  • Reference [34]
  • | |
  • Cited by
  • | |
  • Comments
    Abstract:

    An improved YOLOv8 model (FCU-YOLOv8) is proposed to enhance the accuracy and efficiency of rice disease detection, addressing the challenges of diverse rice diseases, complex backgrounds, and subtle differences in characteristics between diseases. The FasterNeXt module is used to replace the C2f module in the YOLOv8 backbone network. By optimizing the network structure, the FasterNeXt module reduces computation and memory access while improving feature extraction efficiency, thus lowering the inference cost of the model. The C3K module (multi-scale convolution module) and CPSA module (convolutional attention mechanism) are designed to further enhance the model’s ability to perceive disease region features. The C3K module allows the model to adapt to disease characteristics at various scales through flexible convolutional kernel selection, while the CPSA module employs an attention mechanism to enhance the model’s ability to capture key information. To improve the quality of detection boxes and the detection performance of dense disease targets, the optimized unified intersection over union (UIoU) loss function is adopted. This function improves detection performance by balancing the accuracy and consistency of bounding boxes during the regression phase. On a custom-made image dataset of eight common rice diseases, FCU-YOLOv8 demonstrates significant improvements over the original YOLOv8 in several performance metrics. The mAP@0.5 index reaches 94.7%, a 2.4% improvement over the baseline model, and the mAP@0.5:0.95 index reaches 67.2%, a 3.3% improvement. The model’s parameters are reduced by 24.2%, and the calculated floating-point operations decrease by 28.7%, compared to the baseline model in terms of model lightweighting. Compared with mainstream algorithms, the proposed algorithm outperforms current leading algorithms, demonstrating the effectiveness of the network.

    Reference
    [1] Lu Y, Yi S J, Zeng NY, et al. Identification of rice diseases using deep convolutional neural networks. Neurocomputing, 2017, 267: 378–384.
    [2] 朱文杰, 孟鑫, 李根, 等. 水稻病虫害目标检测技术研究进展. 农业工程, 2024, 14(6): 39–46.
    [3] Annabel LSP, Annapoorani T, Deepalakshmi P. Machine learning for plant leaf disease detection and classification—A review. Proceedings of the 2019 International conference on communication and signal processing (ICCSP). Chennai: IEEE, 2019. 538–542.
    [4] Phadikar S, Sil J, Das AK. Classification of rice leaf diseases based on morphological changes. International Journal of Information and Electronics Engineering, 2012, 2(3): 460–463.
    [5] Xu YZ, Yu GZ, Wang YP, et al. Car detection from low-altitude UAV imagery with the faster R-CNN. Journal of Advanced Transportation, 2017, 2017(1): 2823617.
    [6] 周飞燕, 金林鹏, 董军. 卷积神经网络研究综述. 计算机学报, 2017, 40(6): 1229–1251.
    [7] Carion N, Massa F, Synnaeve G, et al. End-to-end object detection with Transformers. Proceedings of the 16th European Conference on Computer Vision. Cham: Springer, 2020. 213–229.
    [8] Zhou GX, Zhang WZ, Chen AB, et al. Rapid detection of rice disease based on FCM-KM and faster R-CNN fusion. IEEE Access, 2019, 7: 143190–143206.
    [9] Bari BS, Islam MN, Rashid M, et al. A real-time approach of diagnosing rice leaf disease using deep learning-based faster R-CNN framework. PeerJ Computer Science, 2021, 7: e432.
    [10] Zhang Z, Jiang L, Hong QQ, et al. Rice disease identification method based on improved mask R-CNN. Proceedings of the 2021 International Conference on Computer Information Science and Artificial Intelligence (CISAI). Kunming: IEEE, 2021. 250–253.
    [11] Yang H, Deng XQ, Shen H, et al. Disease detection and identification of rice leaf based on improved detection Transformer. Agriculture, 2023, 13(7): 1361.
    [12] 严陈慧子, 田芳明, 谭峰, 等. 基于改进YOLOv4的水稻病害快速检测方法. 江苏农业科学, 2023, 51(6): 187–194.
    [13] 周思捷, 刘天奇, 陈天华. 基于改进YOLOv5算法的水稻病害识别研究. 中国农机化学报, 2024, 45(8): 246–253.
    [14] Yuan CC, Liu TH, Gao FY, et al. YOLOv5s-CBAM-DMLHead: A lightweight identification algorithm for weedy rice (Oryza sativa f.spontanea) based on improved YOLOv5. Crop Protection, 2023, 172: 106342.
    [15] Jia LQ, Wang T, Chen Y, et al. MobileNet-CA-YOLO: An improved YOLOv7 based on the MobileNetV3 and attention mechanism for rice pests and diseases detection. Agriculture, 2023, 13(7): 1285.
    [16] 戴林华, 黎远松, 石睿. 基于改进YOLOv8n算法的水稻叶片病害检测. 湖北民族大学学报(自然科学版), 2024, 42(3): 382–388.
    [17] Chen JR, Kao SH, He H, et al. Run, don’t walk: Chasing higher FLOPs for faster neural networks. Proceedings of the 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). Vancouver: IEEE, 2023. 12021–12031.
    [18] Luo XJ, Cai ZH, Shao B, et al. Unified-IoU: For high-quality object detection. arXiv:2408.06636, 2024.
    [19] Sethy PK, Barpanda NK, Rath AK, et al. Deep feature based rice leaf disease identification using support vector machine. Computers and Electronics in Agriculture, 2020, 175: 105527.
    [20] Wang W, Howard A. MOSAIC: Mobile segmentation via decoding aggregated information and encoded context. arXiv:2112.11623, 2021.
    [21] Zhang HY, Cisse M, Dauphin YN, et al. Mixup: Beyond empirical risk minimization. Proceedings of the 6th International Conference on Learning Representations. Vancouver: OpenReview.net, 2018. 1–13.
    [22] Ge Z, Liu ST, Wang F, et al. YOLOX: Exceeding YOLO series in 2021. arXiv:2107.08430, 2021.
    [23] Lin TY, Dollár P, Girshick R, et al. Feature pyramid networks for object detection. Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). Honolulu: IEEE, 2017. 936–944.
    [24] Liu S, Qi L, Qin HF, et al. Path aggregation network for instance segmentation. Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Salt Lake City: IEEE, 2018. 8759–8768.
    [25] Feng CJ, Zhong YJ, Gao Y, et al. TOOD: Task-al-igned one-stage object detection. arXiv:2108.07755, 2021.
    [26] Zheng ZH, Wang P, Ren DW, et al. Enhancing geometric factors in model learning and inference for object detection and instance segmentation. IEEE Transactions on Cybernetics, 2022, 52(8): 8574–8586.
    [27] Liu HJ, Liu FQ, Fan XY, et al. Polarized self-attention: Towards high-quality pixel-wise regression. arXiv:2107.00782, 2021.
    [28] Lin TY, Goyal R, Girshick R, et al. Focal loss for dense object detection. Proceedings of the 2017 IEEE International Conference on Computer Vision (ICCV). Venice: IEEE, 2017. 2999–3007.
    [29] Li CY, Li LL, Jiang HL, et al. YOLOv6: A single-stage object detection framework for industrial applications. arXiv:2209. 02976, 2022.
    [30] Wang CY, Yeh IH, Liao HYM. YOLOv9: Learning what you want to learn using programmable gradient information. Proceedings of the 18th European Conference on Computer Vision. Milan: Springer, 2024. 1–21.
    [31] Wang A, Chen H, Liu LH, et al. YOLOv10: Real-time end-to-end object detection. Proceedings of the 38th Conference on Neural Information Processing Systems. Vancouver: NeurIPS, 2024. 1–28.
    [32] Jocher G, Qiu J, Chaurasia A. Ultralytics YOLO (Version 8.0.0) [Computer software]. https://github.com/ultralytics/ultralytics. [2024-10-12].
    [33] Peng Y, Li H, Wu P, et al. D-FINE: Redefine regression task of DETRs as fine-grained distribution refinement. arXiv:2410.13842, 2024.
    [34] Huang SH, Lu ZC, Cun XD, et al. DEIM: DETR with improved matching for fast convergence. arXiv:2412.04234, 2024.
    Related
    Cited by
    Comments
    Comments
    分享到微博
    Submit
Get Citation

聂俊,朱节中.基于改进YOLOv8的水稻病害检测.计算机系统应用,,():1-14

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:October 24,2024
  • Revised:January 15,2025
  • Online: April 01,2025
Article QR Code
You are the first991371Visitors
Copyright: Institute of Software, Chinese Academy of Sciences Beijing ICP No. 05046678-3
Address:4# South Fourth Street, Zhongguancun,Haidian, Beijing,Postal Code:100190
Phone:010-62661041 Fax: Email:csa (a) iscas.ac.cn
Technical Support:Beijing Qinyun Technology Development Co., Ltd.

Beijing Public Network Security No. 11040202500063