CAFM-Enhanced YOLOv8: A Two-Stage Optimization for Precise Strawberry Disease Detection in Complex Field Conditions

Authors:
Open Access
Publisher website
Google Scholar
Abstract
Strawberry, as an important global economic crop, its disease prevention and control directly affects yield and quality. Traditional detection means rely on manual observation or traditional machine learning algorithms, which have defects such as low efficiency, high false detection rate, and insufficient adaptability to tiny disease spots and complex environment. To solve the above problems, this study proposes a strawberry disease recognition method based on improved YOLOv8. By systematically acquiring 3146 image data covering seven types of typical diseases, such as gray mold and powdery mildew, a high-quality dataset containing different disease stages and complex backgrounds was constructed. Aiming at the difficulties in disease detection, the YOLOv8 model is optimized in two stages: on the one hand, the ultra-small scale detection head (32 × 32) is introduced to enhance the model’s ability to capture early tiny spots; on the other hand, the convolution and attention fusion module (CAFM) is combined to enhance the feature robustness in complex field scenes through the synergy of local feature extraction and global information focusing. Experiments show that the mAP50 of the improved model reaches 0.96 and outperforms mainstream algorithms such as YOLOv5 and Faster R-CNN in both recall and F1 score. In addition, the interactive system developed based on the PyQT5 framework can process images, videos and camera inputs in real time, and the disease areas are presented intuitively through visualized bounding boxes and category labels, which provides farmers with a lightweight and low-threshold field management tool. This study not only verifies the effectiveness of the improved algorithm but also provides a practical reference for the engineering application of deep learning in agricultural scenarios, which is expected to promote the further implementation of precision agriculture technology.
Funding Information
  • Key Laboratory for Crop Production and Smart Agriculture of Yunnan Province (202301BD070001-203, 202101AT070267, 2024ZHNY10, YNAUKCSZJG2023056, 2020XJZK01, 2021J0943)
  • Yunnan Provincial Basic Research Project (202301BD070001-203, 202101AT070267, 2024ZHNY10, YNAUKCSZJG2023056, 2020XJZK01, 2021J0943)
  • The Key Laboratory for Crop Production and Smart Agriculture of Yunnan Province 2024 Open Fund Project (202301BD070001-203, 202101AT070267, 2024ZHNY10, YNAUKCSZJG2023056, 2020XJZK01, 2021J0943)
  • Yunnan Agricultural University Education and Teaching Reform Research Project (202301BD070001-203, 202101AT070267, 2024ZHNY10, YNAUKCSZJG2023056, 2020XJZK01, 2021J0943)
  • Yunnan Agricultural University University-level First-Class Undergraduate Curriculum Project (Big Data Storage and Processing Technology) (202301BD070001-203, 202101AT070267, 2024ZHNY10, YNAUKCSZJG2023056, 2020XJZK01, 2021J0943)
  • Scientific Research Fund Project of Yunnan Provincial Education Department (202301BD070001-203, 202101AT070267, 2024ZHNY10, YNAUKCSZJG2023056, 2020XJZK01, 2021J0943)

This publication has 13 references indexed in Scilit: