基于多尺度融合模块和特征增强的杂草检测方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

陕西省重点研发计划项目(2021GY-022)、西安市科技计划项目(2019216514GXRC001CG002-GXYD1.7)和国家留学基金项目(201708615011)


Weed Detection Based on Multi-scale Fusion Module and Feature Enhancement
Author:
Affiliation:

Fund Project:

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

    针对单步多框检测器(Single shot multibox detector,SSD)网络模型参数多、小目标检测效果差、作物与杂草检测精度低等问题,提出一种基于多尺度融合模块和特征增强的杂草检测方法。首先将轻量网络MobileNet作为SSD模型的特征提取网络,并设计了一种多尺度融合模块,将浅层特征图先通过通道注意力机制增强图像中的关键信息,再将特征图经过不同膨胀系数的扩张卷积扩大感受野,最后将两条分支进行特征融合,对于检测小目标的浅层特征图,在包含较多小目标细节信息的同时,还包含丰富的语义信息。在此基础上对输出的6个特征图经过通道注意力机制进行特征增强。实验结果表明,本文提出的基于多尺度融合模块和特征增强的杂草检测模型,在自然环境下甜菜与杂草图像数据集中,平均检测精度可达88.84%,较标准SSD模型提高了3.23个百分点,参数量减少57.09%,检测速度提高88.44%,同时模型对小目标作物与杂草以及叶片交叠情况的检测能力均有提高。

    Abstract:

    Aiming at the problems of single shot multibox detector (SSD) network model with large parameters, poor detection of small targets and low detection accuracy of crops and weeds, a weed detection method based on multi-scale fusion module and feature enhancement was proposed. Firstly, MobileNet, a lightweight network, was used as the feature extraction network of SSD model to reduce the amount of model parameters and improve the speed of model feature extraction. And a multi-scale fusion module was designed to enhance the key information in the shallow feature map by channel attention mechanism, and then the receptive field was expanded by dilated convolution with different expansion rates. Finally, the two branches were fused, so that the shallow feature map used to detect small targets can contain rich semantic information while containing more detailed information of small targets. On this basis, the output six feature maps were feature enhanced by the channel attention mechanism to enhance the key features in the images and make the extracted features more directional, thus improving the detection accuracy of the model for crops and weeds. The experimental results showed that the weed detection model based on multi-scale fusion module and feature enhancement proposed can achieve an average detection accuracy of 8884% in the image data set of sugar beet and weeds in the natural environment, which was 3.23 percentage points better than that of the standard SSD model, 57.09% less parameters, and 88.44% faster detection speed, while the model's ability to detect small-scale crops and weeds, and leaf overlap were all improved.

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

亢洁,刘港,郭国法.基于多尺度融合模块和特征增强的杂草检测方法[J].农业机械学报,2022,53(4):254-260. KANG Jie, LIU Gang, GUO Guofa. Weed Detection Based on Multi-scale Fusion Module and Feature Enhancement[J]. Transactions of the Chinese Society for Agricultural Machinery,2022,53(4):254-260.

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