基于注意力机制和边缘感知的田梗提取模型
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家重点研发计划项目(2019YFE0125500-04)、国家自然科学基金青年基金项目(32101617)、江苏省农业科技自主创新项目(CX(22)3201)和中国博士后科学基金项目(2022T150327)


Ridge Extraction Model Based on Attention Mechanism and Edge Perception
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    田埂精确提取是数字化农业管理的重要前提。针对由于遮挡、斑秃等因素干扰,给基于语义分割方法提取田埂带来困难问题,提出一种基于注意力机制和边缘感知模块的U-Net网络实现田埂提取。首先,将多信息注意力引入U型分割网络的下采样中,增强相邻层之间的上下文信息,提升对田埂区域语义特征的表示能力。其次,将边缘感知分割模块应用至U-Net解码部分的每一层,在不同语义特征层提取田埂边缘信息,提高田埂区域语义分割精度。最后,联合边缘感知损失与语义分割损失构建联合损失函数,用于整体网络优化。通过对安徽省淮北市濉溪县小麦基地采集的无人机麦田数据集进行训练和模型验证,实验结果表明,本文模型语义分割像素准确率高达95.57%,平均交并比达到77.48%。

    Abstract:

    Accurate extraction of ridges is an important prerequisite for digital agricultural management. However, due to the interference of factors such as occlusion and alopecia areata, it brings challenges for the semantic segmentation method to extract the ridge area. A U-Net segmentation network model was proposed based on a multi-information attention mechanism and an edge-aware module. Firstly, multi-information attention was introduced into the down-sampling of the U-shaped network to enhance the context information between adjacent layers and improve the representation ability of the semantic features of the ridge area. Secondly, the edge-aware segmentation module was applied to each layer of the U-Net decoding part, and the ridge edge information was extracted in different semantic feature layers to improve the semantic segmentation accuracy of the ridge region. Finally, the joint edge-aware loss and semantic segmentation loss were used to construct a joint loss function for overall network optimization. The training and model validation were carried out with the UAV wheat field data set collected by the wheat experimental base in Suixi County, Huaibei City, Anhui Province. The experimental results showed that the pixel accuracy of semantic segmentation of crop plants in different datasets was as high as 95.57%, and the average intersection ratio was 77.48%.

    参考文献
    相似文献
    引证文献
引用本文

顾兴健,刘子儒,任守纲,郑恒彪,徐焕良.基于注意力机制和边缘感知的田梗提取模型[J].农业机械学报,2023,54(5):210-218. GU Xingjian, LIU Ziru, REN Shougang, ZHENG Hengbiao, XU Huanliang. Ridge Extraction Model Based on Attention Mechanism and Edge Perception[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(5):210-218.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-09-24
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-05-10
  • 出版日期: