基于改进U-Net网络的苹果叶部病害语义分割方法 |
| |
引用本文: | 王英允,龙燕,杨智优,黄铝文.基于改进U-Net网络的苹果叶部病害语义分割方法[J].浙江农业学报,2023(11):2731-2741. |
| |
作者姓名: | 王英允 龙燕 杨智优 黄铝文 |
| |
作者单位: | 1. 西北农林科技大学机械与电子工程学院;2. 农业农村部农业物联网重点实验室;3. 陕西省农业信息感知与智能服务重点试验室;4. 西北农林科技大学信息工程学院 |
| |
基金项目: | 陕西省重点研发计划(2020NY-144); |
| |
摘 要: | 针对自然条件下苹果叶部病斑分割与识别效果欠佳的问题,本文提出一种融合条件随机场和卷积块状注意力模块的苹果叶部病害语义分割模型,实现苹果叶部锈病、褐斑病、灰斑病及斑点落叶病的病斑准确分割和识别。本文在U-Net模型基础上,使用ResNet50为骨干网络防止梯度消失问题,并分别在跳跃连接分支与上采样层加入卷积块状注意力模块,减少训练过程中的分割精度损失,融合Dice Loss和Focal Loss降低损失波动,最后利用条件随机场优化分割结果,获取病斑掩模图像,实现对苹果叶部病害语义分割。本研究在自制苹果叶部病害数据集上进行试验,分析了光照、阴影及水滴等因素对分割结果的影响。试验结果表明:本文构建的语义分割模型相比传统U-Net模型,平均分割精度(mIoU)提升8.24百分点,平均分类精度(mPrecision)提升11百分点,类别平均像素准确率(mPA)提升6.09百分点,受光照不均、雨滴的影响更小,具有更好的鲁棒性和可靠性。
|
关 键 词: | 病害分割 注意力机制 条件随机场 深度语义分割 |
|
|