融合通道注意力机制的ResUnet作物病斑分割模型

doi:10.13733/j.jcam.issn.2095-5553.2024.08.033

摘要/Abstract

摘要： 针对温室环境复杂背景下番茄早疫病病斑难以准确识别的问题，提出一种融合通道注意力机制的ResUnet模型。构建温室环境复杂背景下的番茄早疫病数据集；通过融合通道注意力机制的ResUnet模型进行病斑分割，其中ResUnet网络能够学习不同深度特征的重要性，嵌入通道注意力机制使改进的模型更加关注病斑的位置特征。该模型对番茄叶部早疫病病斑分割的准确率为97%，比Unet和Resnet101模型分别提高1.99%和2.97%。将番茄早疫病病斑分割模型得到的骨干网络层参数和权重迁移到单一背景的辣椒结痂、苹果灰斑病、葡萄黑腐病等病斑分割模型，进行改进及参数的微调处理，均能实现病斑的准确分割。在研究算法基础上，设计智能诊断系统，可对作物病害进行快速准确诊断，为及时防控提供依据。

关键词: 作物病斑, 番茄早疫病, ResUnet, 注意力机制, 迁移学习, 智能诊断

Abstract: In view of the difficulty in accurately identifying tomato early blight spot in the complex greenhouse environment, a ResUnet model of integrated channel attention mechanism was proposed. The data set of tomato early blight under complex greenhouse environment was constructed. Lesions were segmented by ResUnet model integrating channel attention mechanism, in which ResUnet network could learn the importance of different depth features, and embedding channel attention mechanism made the improved model pay more attention to the location features of lesions. The accuracy of this model was 97%, 1.99% and 2.97% higher than that of Unet and Resnet101 models, respectively. The parameters and weights of backbone network layer obtained from tomato early blight spot segmentation model were transferred to the single background spot segmentation model of pepper scab, apple gray spot, grape black rot, etc., and improved and the parameters were fine‑tuned to achieve accurate spot segmentation. On the basis of the research algorithm, the intelligent diagnosis system is designed, which can quickly and accurately diagnose the studied crop diseases and provide basis for timely prevention and control.

Key words: leaf disease, tomato early blight disease, ResUnet, attention mechanism, transfer learning, intelligent diagnose

中图分类号:

S225.92

苏斐, 王光辉, 史艳霞, 贾然, 闫银发, 祖林禄, . 融合通道注意力机制的ResUnet作物病斑分割模型[J]. 中国农机化学报, 2024, 45(8): 228-233.

Su Fei, , Wang Guanghui, Shi Yanxia, Jia Ran, Yan Yinfa, , Zu Linlu, . Leaf disease segmentation model of greenhouse tomatoes based on ResUnet with attention mechanism[J]. Journal of Chinese Agricultural Mechanization, 2024, 45(8): 228-233.

参考文献

[ 1 ] Goyal L， Sharma C M， Singh A， et al. Leaf and spike wheat disease detection & classification using an improved deep convolutional architecture [J]. Informatics in Medicine Unlocked， 2021， 25: 100642.
[ 2 ] 李冠林，马占鸿，黄冲，等. 基于K_means硬聚类算法的葡萄病害彩色图像分割方法[J]. 农业工程学报， 2010， 26（S2）: 32-37.
Li Guanlin， Ma Zhanhong， Huang Chong， et al. Segmentation of color images of grape diseases using K_means clustering algorithm [J]. Transactions of the Chinese Society of Agricultural Engineering， 2010， 26（1）: 32-37.
[ 3 ] 王振，张善文，赵保平. 基于级联卷积神经网络的作物病害叶片分割[J]. 计算机工程与应用， 2020， 56（15）: 242-250.
Wang Zhen， Zhang Shanwen， Zhao Baoping. Crop diseases leaf segmentation method based on cascade convolutional neural network [J]. Computer Engineering and Applications， 2020， 56（15）: 242-250.
[ 4 ] 赵兵，冯全. 基于全卷积网络的葡萄病害叶片分割[J]. 南京农业大学学报， 2018， 41（4）: 752-759.
Zhao Bing， Feng Quan. Segmentation of grape diseases leaf based on full convolution network [J]. Journal of Nanjing Agricultural University， 2018， 41（4）: 752-759.
[ 5 ] Wang C， Du P， Wu H， et al. A cucumber leaf disease severity classification method based on the fusion of DeepLabV3+ and U-Net [J]. Computers and Electronics in Agriculture， 2021， 106373: 0168-1699.
[ 6 ] 赵小虎，李晓，叶圣，等. 基于改进U-Net网络的多尺度番茄病害分割算法[J]. 计算机工程与应用， 2022， 58（10）: 216-223.
Zhao Xiaohu， Li Xiao， Ye Sheng， et al. Multi‑scale tomato disease segmentation algorithm based on improved U-Net network [J]. Computer Engineering and Applications， 2022， 58（10）: 216-223.
[ 7 ] 任守纲，贾馥玮，顾兴健，等. 反卷积引导的番茄叶部病害识别及病斑分割模型[J]. 农业工程学报， 2020， 36（12）: 186-195.
Ren Shougang， Jia Fuwei， Gu Xingjian， et al. Recognition and segmentation model of tomato leaf diseases based on deconvolution‑guiding [J]. Transactions of the Chinese Society of Agricultural Engineering， 2020， 36（12）: 186-195.
[ 8 ] Chen J， Zhang D， Zeb A， et al. Identification of rice plant diseases using lightweight attention networks [J]. Expert Systems with Applications， 2021， 169: 114514.
[ 9 ] Jie H， Li S， Sun G. Squeeze-and-excitation networks [J]. IEEE Transactions on Pattern Analysis and Machine Intelligence， 2018: 7132-7141.
[10] 苏宝峰，沈磊，陈山. 基于注意力机制的葡萄品种多特征分类方法[J]. 农业机械学报， 2021， 52（11）: 226-233， 252.
Su Baofeng， Shen Lei， Chen Shan， et al. Multi‑features identification of grape cultivars based on attention mechanism [J]. Transactions of the Chinese Society for Agricultural Machinery， 2021， 52（11）: 226-233， 252.
[11] 赵晋陵，詹媛媛，王娟，等. 基于SE-UNet的冬小麦种植区域提取方法[J]. 农业机械学报， 2022， 53（9）: 189-196.
Zhang Jinling， Zhan Yuanyuan， Wang Juan， et al. SE-UNet-based extraction of winter wheat planting areas [J]. Transactions of the Chinese Society for Agricultural Machinery， 2022， 53（9）: 189-196.
[12] Zeng W， Li M. Crop leaf disease recognition based on Self‑Attention convolutional neural network [J]. Computers and Electronics in Agriculture， 2020， 172: 10534.

[1]	陈浩宇, 苗玉彬. 基于GASF变换和深度学习的柑橘内部品质分析[J]. 中国农机化学报, 2025, 46(4): 133-138.
[2]	赵玉清, 焦雨杰, 李宏, 王天允, 李嘉舜, 张悦, . 改进ShuffleNet V1算法的咖啡豆分级方法研究[J]. 中国农机化学报, 2025, 46(4): 194-203.
[3]	马瑞峻, 丁世春, 陈瑜. 基于改进MobileNetV3—Small的甘薯外部品质分类方法#br#[J]. 中国农机化学报, 2025, 46(4): 211-217.
[4]	王丹阳, 梁伟红, 李玉萍, 黄贵修, . 面向不平衡数据的木薯叶部病害图像识别方法[J]. 中国农机化学报, 2025, 46(3): 101-107.
[5]	吕宗旺, 王甜甜, 孙福艳, 祝玉华, . 基于改进YOLOv8m的小麦仓储粮虫检测方法[J]. 中国农机化学报, 2025, 46(3): 108-114.
[6]	刘起, 代东南, 孙孟研, 马德新, 徐阳. 基于双重注意力融合网络结构的圣女果果实表面缺陷识别[J]. 中国农机化学报, 2025, 46(3): 153-159.
[7]	沈中华, 程虎强, , 夏爱强, 李涵. 基于RGE—UNet模型的甘蔗蔗梢识别研究[J]. 中国农机化学报, 2025, 46(3): 188-194.
[8]	郭敬涛, 吕凤, 章慧婷, 杨彪, 刘大洋. 基于迁移学习与轻量化YOLOv5s的草莓目标检测方法[J]. 中国农机化学报, 2025, 46(3): 253-260.
[9]	颜士军, 朱红梅, 王雅童, 张亮. 基于字词融合和注意力机制的兽药文本命名实体识别#br#[J]. 中国农机化学报, 2025, 46(3): 336-342.
[10]	吴秋兰, 陈雪飞, 陈超, 张峰, 王姝妹, 赵恒. 基于改进YOLOv5s的香菇菌棒污染识别方法[J]. 中国农机化学报, 2025, 46(2): 217-223.
[11]	戴敏, 孙文靖, 缪宏. 基于轻量化CBAM—GoogLeNet的辣椒病虫害识别[J]. 中国农机化学报, 2025, 46(2): 224-229.
[12]	李涛, 买买提明⋅艾尼, 古丽巴哈尔⋅托乎提, 杨佳雨. 基于改进YOLOv8的小棚架下无核白葡萄果梗识别[J]. 中国农机化学报, 2025, 46(2): 259-263.
[13]	段小勇, 何超, 刘学渊. 基于改进DeepLabV3+的非结构化道路可行驶区域检测[J]. 中国农机化学报, 2025, 46(2): 271-278.
[14]	李先旺, 刘赛虎, 黄忠祥, 章霞东. 基于XLNet—BiLSTM—AFF—CRF的谷物收割机械维修知识命名实体识别[J]. 中国农机化学报, 2025, 46(2): 319-325.
[15]	高芳征, 温鑫, 黄家才, 陈光明, 金少宇, 赵雪迪. 基于AD-YOLOX-Nano的茶叶嫩芽识别算法[J]. 中国农机化学报, 2025, 46(1): 178-184.