数字农科院2.0

An improved YOLOv5 model based on visual attention mechanism: Application to recognition of tomato virus disease

文献类型： 外文期刊

作者： Jiangtao Qi;Xiangnan Liu;Kai Liu;Farong Xu;Hui Guo;Xinliang Tian;Mao Li;Zhiyuan Bao;Yang Li

作者机构：

关键词： Attention mechanism;Disease detection;SE-YOlOv5 model;Tomato virus disease

期刊名称： Computers and Electronics in Agriculture

ISSN： 0168-1699

年卷期： 2022 年 194 卷

页码：

收录情况： SCIE（2022版） ; EI（2022版）

摘要： Traditional target detection methods cannot effectively screen key features, which leads to overfitting and produces a model with a weak generalization ability. In this paper, an improved SE-YOLOv5 network model is proposed for the recognition of tomato virus diseases. Images of tomato diseases in greenhouses were collected using a mobile phone, and the collected images were expanded. A squeeze-and-excitation (SE) module was added to a YOLOv5 model to realize the extraction of key features, using a human visual attention mechanism for reference. The trained network model was evaluated on the test set of tomato virus diseases. The accuracy was 91.07%, which was 7.12%, 17.85% and 8.91% higher than that of the Faster regions with convolutional neural network features (R-CNN) model, single-shot multiBox detector (SSD) model and YOLOv5 model, respectively. Meanwhile, the mean average precision (mAP@0.5) was 94.10%, which was 1.23%, 16.77% and 1.78% higher than that of the Faster R-CNN model, SSD model and YOLOv5 model. The proposed SE-YOLOv5 model can effectively detect regions of tomato virus disease, which provides disease identification and control theoretical research and technical support.

分类号：

  • 相关文献

[1]Pseudo high-frequency boosts the generalization of a convolutional neural network for cassava disease detection. Jiayu Zhang,Chao Qi,Peter Mecha,Yi Zuo,Zongyou Ben,Haolu Liu,Kunjie Chen. 2022

[2]Xoo-YOLO: a detection method for wild rice bacterial blight in the field from the perspective of unmanned aerial vehicles. Pan Pan,Wenlong Guo,Xiaoming Zheng,Lin Hu,Guomin Zhou,Jianhua Zhang. 2023

[3]Rectifying the extremely weakened signals for cassava leaf disease detection. Jiayu Zhang,Baohua Zhang,Innocent Nyalala,Peter Mecha,Junlong Chen,Kunjie Chen,Junfeng Gao. 2025

[4]Spectroscopic detection of cotton Verticillium wilt by spectral feature selection and machine learning methods. Li, Weinan,Liu, Lisen,Li, Jianing,Yang, Weiguang,Guo, Yang,Huang, Longyu,Yang, Zhaoen,Peng, Jun,Jin, Xiuliang,Lan, Yubin. 2025

[5]Apnet: Lightweight network for apricot tree disease and pest detection in real-world complex backgrounds. Li, Minglang,Tao, Zhiyong,Yan, Wentao,Lin, Sen,Feng, Kaihao,Zhang, Zeyi,Jing, Yurong. 2025

[6]Identification of Pine Wilt Disease Infected Wood Using UAV RGB Imagery and Improved YOLOv5 Models Integrated with Attention Mechanisms. Peng Zhang,Zhichao Wang,Yuan Rao,Jun Zheng,Ning Zhang,Degao Wang,Jianqiao Zhu,Yifan Fang,Xiang Gao. 2023

[7]Noise-tolerant RGB-D feature fusion network for outdoor fruit detection. Qixin Sun,Xiujuan Chai,Zhikang Zeng,Guomin Zhou,Tan Sun. 2022

[8]A lightweight tea bud detection model based on Yolov5. Gui, Zhiyong,Chen, Jianneng,Li, Yang,Chen, Zhiwei,Wu, Chuanyu,Dong, Chunwang. 2023

[9]Identification method of vegetable diseases based on transfer learning and attention mechanism. Xue Zhao,Kaiyu Li,Yunxia Li,Juncheng Ma,Lingxian Zhang. 2022

[10]Attention-optimized DeepLab V3+for automatic estimation of cucumber disease severity. Li, Kaiyu,Zhang, Lingxian,Li, Bo,Li, Shufei,Ma, Juncheng. 2022

[11]Automatic rape flower cluster counting method based on low-cost labelling and UAV-RGB images. Jie Li,Enguo Wang,Jiangwei Qiao,Yi Li,Li Li,Jian Yao,Guisheng Liao. 2023

[12]Automatic counting of rapeseed inflorescences using deep learning method and UAV RGB imagery. Li J.,Li Y.,Qiao J.,Li L.,Wang X.,Yao J.,Liao G.. 2023

[13]Sunflower-YOLO: Detection of sunflower capitula in UAV remote sensing images. Rui Jing,Qinglin Niu,Yuyu Tian,Heng Zhang,Qingqing Zhao,Zongpeng Li,Xinguo Zhou,Dongwei Li. 2024

[14]Semi-supervised learning and attention mechanism for weed detection in wheat. Teng Liu,Xiaojun Jin,Luyao Zhang,Jie Wang,Yong Chen,Chengsong Hu,Jialin Yu. 2023

[15]Automatic semantic modeling of structured data sources with cross-modal retrieval. Xu R.,Mayer W.,Chu H.,Zhang Y.,Zhang H.-Y.,Wang Y.,Liu Y.,Feng Z.. 2024

[16]TBTA-D2Net: a novel hyperspectral image classification method based on triple-branch ternary-attention mechanism and Dense2Net. Ting Tang,Shengwei Zhang,Shaopeng Liu,Weihong Yan,Xin Pan. 2023

[17]Multi-Scale Mixed Attention Tea Shoot Instance Segmentation Model. Chen, Dongmei,Cao, Peipei,Yan, Lijie,Chen, Huidong,Lin, Jia,Li, Xin,Yuan, Lin,Wu, Kaihua. 2024

[18]An Unstructured Orchard Grape Detection Method Utilizing YOLOv5s. Wang, Wenhao,Shi, Yun,Liu, Wanfu,Che, Zijin. 2024

[19]Small target tea bud detection based on improved YOLOv5 in complex background. Mengjie Wang,Yang Li,Hewei Meng,Zhiwei Chen,Zhiyong Gui,Yaping Li,Chunwang Dong. 2024

[20]ESG-YOLO: A Method for Detecting Male Tassels and Assessing Density of Maize in the Field. Wu, Wendi,Zhang, Jianhua,Zhou, Guomin,Zhang, Yuhang,Wang, Jian,Hu, Lin. 2024

作者其他论文 更多>>