一种基于改进U-Net的植物图像分割算法

作者：赵艳杰; 郭晓丽; 刘洋^*; 娜茜泰; 李雅婷; 张铭梓; 秦文强

来源：中国传媒大学学报(自然科学版), 2021, 28(03): 32-40.

DOI：10.16196/j.cnki.issn.1673-4793.2021.03.005

• 图像分割
• U-Net
• 残差网络
• 注意力机制

图像分割技术是图像分析的重要步骤。本文针对野外采集的植物图片存在多尺度及理想数据量不足等问题,提出一种多尺度Res-Att-Unet模型以实现自主植物图像分割。该方法在U-Net的收缩网络中增加多尺度机制,在输出端以相应的Label图进行深监督,并在模型的扩张网络部分加入双通道注意力机制,以抑制无关信息表达,各层新增的残差块使得模型可以提取更加丰富的特征信息,实验表明无论在花朵型图像还是叶片型图像中均取得不错的分割效果。

• 单位
  电子信息工程学院; 内蒙古大学