• • 下一篇
李雨希1,王雅莹1,田诗芹2,郑伯川1,3
LI Yuxi1, WANG Yaying1, TIAN Shiqin2, ZHENG Bochuan1,3
摘要: 基于卷积神经网络(CNN)与Transformer的谷穗计数方法仍存在长距离建模困难、计算效率低以及密度误报与密度漏报的问题。为解决这些问题,提出名为PCLMamba的新计数方法。PCLMamba是一个基于分割引导与频域增强的轻量Mamba网络。该网络通过引入分组的多方向状态空间模型(GDMM)分组处理图像序列且并行地送入不同顺序的选择性扫描中,轻量且高效地捕获长距离关系。然后利用通道注意力融合模块(CAFM)融合局部特征与全局扫描特征。为了解决密度误报与密度漏报的问题,设计了分割模块(Segmenter)提供先验知识,以抑制背景和增强前景。同时设计了双域信息增强模块(FSE),该模块将中间特征在空域与频域上进行处理增强,进一步提高计数精度。实验结果表明,PCLMamba在MTC、WED、SHC数据集的计数任务上表现出色,平均绝对误差(MAE)分别达到4.4、4.2、15.4,且其他指标也整体优于对比方法。
中图分类号: