Suppr超能文献

基于混合损失函数和CBAM的苹果叶片小病斑分割识别研究

Research of segmentation recognition of small disease spots on apple leaves based on hybrid loss function and CBAM.

作者信息

Zhang Xiaoqian, Li Dongming, Liu Xuan, Sun Tao, Lin Xiujun, Ren Zhenhui

机构信息

College of Mechanical and Electrical Engineering, Hebei Agricultural University, Baoding, China.

出版信息

Front Plant Sci. 2023 Jun 6;14:1175027. doi: 10.3389/fpls.2023.1175027. eCollection 2023.

Abstract

Identification technology of apple diseases is of great significance in improving production efficiency and quality. This paper has used apple Alternaria blotch and brown spot disease leaves as the research object and proposes a disease spot segmentation and disease identification method based on DFL-UNet+CBAM to address the problems of low recognition accuracy and poor performance of small spot segmentation in apple leaf disease recognition. The goal of this paper is to accurately prevent and control apple diseases, avoid fruit quality degradation and yield reduction, and reduce the resulting economic losses. DFL-UNet+CBAM model has employed a hybrid loss function of Dice Loss and Focal Loss as the loss function and added CBAM attention mechanism to both effective feature layers extracted by the backbone network and the results of the first upsampling, enhancing the model to rescale the inter-feature weighting relationships, enhance the channel features of leaf disease spots and suppressing the channel features of healthy parts of the leaf, and improving the network's ability to extract disease features while also increasing model robustness. In general, after training, the average loss rate of the improved model decreases from 0.063 to 0.008 under the premise of ensuring the accuracy of image segmentation. The smaller the loss value is, the better the model is. In the lesion segmentation and disease identification test, MIoU was 91.07%, MPA was 95.58%, F1 Score was 95.16%, MIoU index increased by 1.96%, predicted disease area and actual disease area overlap increased, MPA increased by 1.06%, predicted category correctness increased, F1 Score increased by 1.14%, the number of correctly identified lesion pixels increased, and the segmentation result was more accurate. Specifically, compared to the original U-Net model, the segmentation of Alternaria blotch disease, the MIoU value increased by 4.41%, the MPA value increased by 4.13%, the Precision increased by 1.49%, the Recall increased by 4.13%, and the F1 Score increased by 2.81%; in the segmentation of brown spots, MIoU values increased by 1.18%, MPA values by 0.6%, Precision by 0.78%, Recall by 0.6%, and F1 Score by 0.69%. The spot diameter of the Alternaria blotch disease is 0.2-0.3cm in the early stage, 0.5-0.6cm in the middle and late stages, and the spot diameter of the brown spot disease is 0.3-3cm. Obviously, brown spot spots are larger than Alternaria blotch spots. The segmentation performance of smaller disease spots has increased more noticeably, according to the quantitative analysis results, proving that the model's capacity to segment smaller disease spots has greatly improved. The findings demonstrate that for the detection of apple leaf diseases, the method suggested in this research has a greater recognition accuracy and better segmentation performance. The model in this paper can obtain more sophisticated semantic information in comparison to the traditional U-Net, further enhance the recognition accuracy and segmentation performance of apple leaf spots, and address the issues of low accuracy and low efficiency of conventional disease recognition methods as well as the challenging convergence of conventional deep convolutional networks.

摘要

苹果病害识别技术对于提高生产效率和品质具有重要意义。本文以苹果斑点落叶病和褐斑病叶片为研究对象,提出一种基于DFL-UNet+CBAM的病斑分割与病害识别方法,以解决苹果叶片病害识别中识别准确率低、小斑点分割性能差的问题。本文的目标是准确防治苹果病害,避免果实品质下降和产量降低,减少由此造成的经济损失。DFL-UNet+CBAM模型采用Dice Loss和Focal Loss的混合损失函数作为损失函数,并在主干网络提取的有效特征层和第一次上采样结果中均添加了CBAM注意力机制,增强模型重新调整特征间加权关系的能力,增强叶片病斑的通道特征并抑制叶片健康部分的通道特征,提高网络提取病害特征的能力同时增加模型鲁棒性。总体而言,训练后,改进模型在保证图像分割准确率的前提下,平均损失率从0.063降至0.008。损失值越小,模型越好。在病斑分割与病害识别测试中,交并比(MIoU)为91.07%,平均像素精度(MPA)为95.58%,F1分数为95.16%,MIoU指标提高了1.96%,预测病害面积与实际病害面积的重叠度增加,MPA提高了1.06%,预测类别正确率提高,F1分数提高了1.14%,正确识别的病斑像素数量增加,分割结果更准确。具体而言,与原始U-Net模型相比,在斑点落叶病的分割中,MIoU值提高了4.41%,MPA值提高了4.13%,精确率提高了1.49%,召回率提高了4.13%,F1分数提高了2.81%;在褐斑病的分割中,MIoU值提高了1.18%,MPA值提高了0.6%,精确率提高了0.78%,召回率提高了0.6%,F1分数提高了0.69%。斑点落叶病病斑直径在前期为0.2 - 0.3厘米,中后期为0.5 - 0.6厘米,褐斑病病斑直径为0.3 - 3厘米。显然,褐斑病斑比斑点落叶病斑大。根据定量分析结果,较小病斑的分割性能提升更为显著,证明该模型分割较小病斑的能力有了很大提高。研究结果表明,对于苹果叶片病害的检测,本研究提出的方法具有更高的识别准确率和更好的分割性能。与传统U-Net相比,本文模型能够获得更复杂的语义信息,进一步提高苹果叶斑的识别准确率和分割性能,解决传统病害识别方法准确率低、效率低以及传统深度卷积网络收敛困难的问题。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/0ce6/10279884/f260f7c0c007/fpls-14-1175027-g001.jpg

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍。

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索

文档翻译

学术文献翻译模型,支持多种主流文档格式。

立即体验