Journal of Henan Agricultural Sciences ›› 2023, Vol. 52 ›› Issue (9): 156-163.DOI: 10.15933/j.cnki.1004-3268.2023.09.016

• Agricultural Information and Engineering Agricultural Product... • Previous Articles     Next Articles

Target Detection of Pasture Weeds Based on Improved DINO

ZHONG Bin,YANG Jun,LIU Yi,REN Jintao   

  1. (School of Software,Jiangxi Agricultural University,Nanchang 330045,China)
  • Received:2023-06-05 Published:2023-09-15 Online:2023-10-11

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

钟斌,杨珺,刘毅,任锦涛   

  1. (江西农业大学软件学院,江西 南昌 330045)
  • 通讯作者: 杨珺(1970-),男,江西上饶人,副教授,博士,主要从事人工智能、数据挖掘,农业信息化研究。E-mail:ycjum515@163.com
  • 作者简介:钟斌(1999-),男,江苏南京人,在读硕士研究生,研究方向:计算机视觉、特征融合及农业信息化。E-mail:1543541685@qq.com
  • 基金资助:
    江西省自然科学基金面上项目(20212BAB205009)

Abstract: Aiming at the deficiency of low accuracy of traditional weed identification methods for weeds with similar characteristics to forage,an improved DINO detection network for weed detection in pastures was proposed.The CBAM‑G attention mechanism module combining spatial attention and channel attention was added to the backbone network ResNet of the end‑to‑end model DINO to enhance the extraction of effective features and precise positional information,and reduce the interference of invalid information;By increasing the network depth,the backbone network could extract deeper target features;The original feature fusion module in the backbone network was replaced with the lightweight SFPN module;Finally,in order to improve the stability and detection performance of the feature extraction network and Transformer,Gaussian error linear unit was added to the model feature extraction network.The experiment showed that the AP50 of the algorithm on the pasture weed data set reached 95.89%,AP75reached 89.23%. Compared to the original model,it can better utilize multi‑scale feature information and improve recognition accuracy.

Key words: Weed identification, DINO, ResNet, Feature fusion, Transformer

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

关键词: 杂草识别, DINO, ResNet, 特征融合, Transformer

CLC Number: