基于改进DINO的牧场杂草检测

摘要：

针对传统杂草识别方法对与牧草有相似特征的杂草的识别精度低的不足,提出一种基于改进DINO检测网络的牧场杂草检测模型.为了增强有效特征和精确位置信息的提取,并减少无效信息的干扰,在端到端模型DINO的主干网络ResNet中加入结合空间注意力和通道注意力的CBAM-G注意力机制模块;通过增加网络深度,让主干网络可以提取到更深层次的目标特征;引入更加轻量化的SFPN模块,替换了算法中原有的特征融合模块;最后为了提高特征提取网络与Transformer的稳定度和检测性能,在模型特征提取网络中加入高斯误差线性单元.结果表明,改进后的检测模型在Kaggle的牧场杂草数据集上的像素精度AP50达到了 95.89％,AP75达到了 89.23％,相较于原始模型可以更好地利用多尺度特征信息,并提升识别精度.

关键词：杂草识别、DINO、ResNet、特征融合、Transformer

所属期刊栏目：52

分类号：S126;TP391.4(农业物理学)

资助基金：江西省自然科学基金面上项目20212BAB205009

在线出版日期：2023-10-23（万方平台首次上网日期，不代表论文的发表时间）

页数：共8页

页码：156-163