首页|基于改进型UNet语义分割模型的马铃薯病害检测方法

基于改进型UNet语义分割模型的马铃薯病害检测方法

扫码查看
深度学习技术和卷积神经网络的发展,为农作物病害快速准确检测提供了新的解决方法。本文实地采集马铃薯图像,使用UNet语义分割模型对马铃薯病害进行检测,使用了 2 种主干网络模型VGG16 和ResNet50,它们的精确率为 93。00%,F1 为 92。48%和 92。77%,MPA为 94。47%和 94。42%,MIoU为 84。79%和 84。75%。提出了一种改进型UNet语义分割模型,通过在网络的第一次上采样处加入注意力机制模块获取特征图,将注意力机制获得的特征图乘以原输入的特征图进行下一步的上采样网络过程,最终取得的Precision、F1、MPA和MIoU分别为 94。83%、92。89%、95。96%和 86。32%。相较于初始网络的指标有所提高,为自然环境下马铃薯叶片病害识别和检测提供较为全面的深度学习算法和模型研究基础。
Potato disease detection based on improved UNet semantic segmentation model
The development of deep learning technology and convolutional neural network has provided a new solution for the rapid and accurate detection of crop diseases.In this paper,potato images were collected in the field,and the UNet semantic segmentation model was used to detect potato diseases.Two backbone network models VGG16 and ResNet50 were used,whose precision was 93.00%,F1 was 92.48%and 92.77%,and MPA was 94.47%and 94.42%,MIoU was 84.79%and 84.75%.An improved UNet semantic segmentation model was proposed.The feature map was obtained by adding an attention mechanism module at the first upsampling of the network,and the feature map obtained by the attention mechanism was multiplied by the original input feature map for the next step.During the sampling network process,the final Precision,F1,MPA and MIoU were 94.83%,92.89%,95.96%and 85.32%,respectively.Compared with the initial network,the index was improved,which provided a more comprehensive deep learning algorithm and model research basis for the identification and detection of potato leaf diseases in natural environment.

potato leaf diseasesemantic segmentationattention mechanismconvolutional neural network

李旭东、王林柏、张博、孙磊、刘景艳、范晓飞

展开 >

河北农业大学 机电工程学院,河北 保定 071000

马铃薯叶片病害 语义分割 注意力机制 卷积神经网络

2024

河北农业大学学报
河北农业大学

河北农业大学学报

CSTPCD北大核心
影响因子:0.475
ISSN:1000-1573
年,卷(期):2024.47(6)