基于空间和通道注意力的花椒图像枝条分割

来源：计算机应用与软件杂志2025年第3期北京时间：

作者：刘兴科, 杨前, 罗建桥, 刘雪垠, 李柏林

摘要：提取主干枝条是花椒视觉采摘的关键问题。针对现有网络模型难以从复杂自然场景中分割出近景主干枝条的问题，向网络中加入新设计的注意力模块，提出基于空间和通道注意力的RGBD图像花椒枝条分割模型。首先，由两个编码器分别提取彩色和深度图像特征。然后，设计空间注意力模块融合这两种特征。空间权值同时考虑外观和距离信息，能够抑制远景枝条的干扰。最后，构造通道注意力模块关注枝条主要形状信息，避免枝条细节轮廓的干扰。实验表明，相比现有双编码器网络，所提模型的分割准确率、交并比分别提高了14.72%、17.65%。所提模型能够分割出满意的近景主干枝条。

关键词：花椒视觉采摘, RGBD图像分割, 深度学习, 注意力机制