高级检索+

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

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

  • 摘要: 针对单步多框检测器(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.

     

/

返回文章
返回