YOLOv7-GCA: A Lightweight and High-Performance Model for Pepper Disease Detection

文献类型： 外文期刊

第一作者： Yue, Xuejun

作者： Yue, Xuejun;Li, Haifeng;Song, Qingkui;Zeng, Fanguo;Zheng, Jianyu;Ding, Ziyu;Kang, Gaobi;Cai, Yulin;Lin, Yongda;Xu, Xiaowan;Yu, Chaoran;Xu, Xiaowan;Yu, Chaoran

作者机构：

关键词： pepper diseases; YOLOv7-GCA; lightweight; attention mechanism; CFNet

期刊名称：AGRONOMY-BASEL （ 影响因子：3.7； 五年影响因子：4.0 ）

ISSN：

年卷期： 2024 年 14 卷 3 期

页码：

收录情况： SCI

摘要： Existing disease detection models for deep learning-based monitoring and prevention of pepper diseases face challenges in accurately identifying and preventing diseases due to inter-crop occlusion and various complex backgrounds. To address this issue, we propose a modified YOLOv7-GCA model based on YOLOv7 for pepper disease detection, which can effectively overcome these challenges. The model introduces three key enhancements: Firstly, lightweight GhostNetV2 is used as the feature extraction network of the model to improve the detection speed. Secondly, the Cascading fusion network (CFNet) replaces the original feature fusion network, which improves the expression ability of the model in complex backgrounds and realizes multi-scale feature extraction and fusion. Finally, the Convolutional Block Attention Module (CBAM) is introduced to focus on the important features in the images and improve the accuracy and robustness of the model. This study uses the collected dataset, which was processed to construct a dataset of 1259 images with four types of pepper diseases: anthracnose, bacterial diseases, umbilical rot, and viral diseases. We applied data augmentation to the collected dataset, and then experimental verification was carried out on this dataset. The experimental results demonstrate that the YOLOv7-GCA model reduces the parameter count by 34.3% compared to the YOLOv7 original model while improving 13.4% in mAP and 124 frames/s in detection speed. Additionally, the model size was reduced from 74.8 MB to 46.9 MB, which facilitates the deployment of the model on mobile devices. When compared to the other seven mainstream detection models, it was indicated that the YOLOv7-GCA model achieved a balance between speed, model size, and accuracy. This model proves to be a high-performance and lightweight pepper disease detection solution that can provide accurate and timely diagnosis results for farmers and researchers.

分类号：

  • 相关文献
作者其他论文 更多>>

微信小程序