本平台为互联网非涉密平台,严禁处理、传输国家秘密或工作秘密

增强型双重Unet(ED-Unet)烟叶病斑分割模型的建立

Establishment of an enhanced dual Unet(ED-Unet)model for tobacco leaf spot segmentation

  • 摘要: 烟草病害的识别与诊断是烟草病害科学防治的前提和基础。为克服传统深度学习方法分割烟叶病斑时存在的泛化性弱、对噪声敏感等局限性问题,以角斑病、赤星病、野火病、蛙眼病4种病害为研究对象,基于Unet,通过改进特征提取模块、嵌入注意力机制等方法构建了增强型双重Unet(Enhanced Dual Unet,ED-Unet)模型对烟叶病斑进行精准分割。结果表明:①ED-Unet在病斑CPA、Recall、IoU、F1分数指标上分别达到92.75%、90.94%、84.93%、91.81%,整体Dice分数达到94.67%;②ED-Unet模型的参数量、浮点运算量、单幅图像推理时间分别为46.5 M、233.92 GFLOPs、65.096 ms;③相较于Unet、PSP、DeepLab v3+、FCN、SegNet、UNET++、DoubleU-Net,ED-Unet模型的精度取得了显著提升,且较好地控制了模型复杂度,综合性能达到最优。该方法可为烟叶病斑以及其他植物病害的精准分割提供技术支持。

     

    Abstract: The identification and diagnosis of tobacco diseases are the prerequisite and foundation for scientific prevention and control of the diseases. To break the barriers of traditional deep learning methods in segmenting tobacco leaf spots such as weak generalization and sensitivity to noise, four leaf diseases including angular spot, brown spot, wildfire disease and frog eye spot were studied, and the Enhanced Dual Unet(ED-Unet)model based on Unet was constructed to accurately segment these leaf spots by improving the feature extraction module and embedding an attention mechanism. The results showed that: 1)ED-Unet reached 92.75%, 90.94%, 84.93% and 91.81% in indexes of spot CPA, Recall, IoU and F1 score, respectively, with an overall Dice score of 94.67%. 2) The numbers of parameters and floating-point operations and the inference time for a single image of the ED-Unet model were 46.5 M, 233.92 GFLOPs and 65.096 ms, respectively. 3)The accuracy of the ED-Unet model was significantly improved compared with those of Unet, PSP, DeepLab v3 +, FCN, SegNet, UNET ++ and DoubleU-Net, and the ED-Unet model complexity was well controlled, thereby achieving optimal overall performance. This method provides technical supports for the precise segmentation of leaf disease spots of tobacco and other plants.

     

/

返回文章
返回