改进MobileViT网络识别轻量化田间杂草

    Light weight recognition of weeds in the field based on improved MobileViT network

    • 摘要: 为解决现有的田间杂草识别方法无法兼顾高准确率与实时性的问题,该研究提出了一种基于改进MobileViT网络的轻量化杂草识别方法。首先,该方法使用高效的MobileViT网络构建杂草特征提取网络,在保证特征提取能力的同时实现更少的模型参数量与计算量;其次,通过高效通道注意力机制加强下采样后特征图中的重要特征,进一步提升模型的特征提取能力;最后,特征提取网络中的MobileViT模块被用于同时学习局部语义信息和全局语义信息,仅通过少量模块的组合便能够准确地捕捉到不同类别杂草与作物间细微差异。为验证该方法的有效性,该研究以农田环境下采集的玉米幼苗及其4类伴生杂草图像为数据进行了模型训练,试验结果表明,该方法的识别准确率、精准度、召回率和F1分数分别为99.61%、99.60%、99.58%和99.59%,优于VGG-16、ResNet-50、DenseNet-161、MobileNetv2等常用卷积神经网络;同时,可视化结果表明该方法能够有效提取杂草图像中的关键特征,并抑制背景区域对识别结果的影响。该研究提出的方法能够精准、快速地区分出农田环境下形态相似的多种杂草与作物,可为智能除草设备中的杂草识别系统设计提供参考。

       

      Abstract: Weeds in the field are serious threats to agricultural production. The rapid and accurate identification of weed species can be critical steps to automated weed control. However, previous weed recognition relies mainly on hand-designed features, such as shape and texture. Fortunately, computer vision can be expected to serve as intelligent weed control equipment. The performance can also be obtained better on the data samples with significant morphological differences. But it cannot accurately identify the weeds and crops in field environments. It is highly required to promote the expression capabilities of the weed species during recognition in real time, due to the particularity of field weeding tasks. In this study, a novel weed recognition was proposed to balance the high accuracy and real-time performance using the improved lightweight MobileViT network. The efficient Channel Attention (ECA) module was also combined to enhance the important parts of the features for better extraction. Firstly, the MobileViT network structure was selected as the lightweight model using Transformers. Specifically, the MobileViT backbone network was used to replace the self-attention layer in the ViT model with the lightweight separable convolution layer. Some optimization techniques were also introduced, such as the depth separable convolution and channel attention mechanism. The feature extraction was then achieved at a high inference speed, but a lower model size and computational cost. As such, the core network was established for lightweight weed identification. Secondly, the ECA mechanism was used as the learnable channel weights to calculate the importance of each channel, with emphasis on the important channels in the input features. The important features were enhanced to further improve the extraction capability from the down-sampled map. Finally, the MobileViT module was trained to simultaneously learn both local and global semantic information in the feature extraction network using both MobileViT and MobileNet networks. Only a small number of modules were combined to accurately capture the subtle differences between the different classes of weeds and crops, in order to fully meet the requirements of real-time performance and recognition accuracy. The comparative experiments were conducted to verify the effectiveness of the model. The images of corn seedlings and four types of companion weeds were also collected in the field environment. The results show that the recognition accuracy, precision, recall, and F1 score of the improved model were 99.61%, 99.60%, 99.58%, and 99.59%, respectively, indicating better performance than the lightweight convolutional neural networks (CNNs), such as ShuffleNet and MobileNet, and the general CNNs, such as VGG-16, ResNet-50, and DenseNet-161. Furthermore, the inference time was only 83ms for the single image, indicating that the recognition efficiency fully met the real-time requirement for the weed identification during weeding operations. The visualization also showed that the key features were effectively extracted from the weed images without the influence of background areas. The new model can be expected to accurately and rapidly distinguish the multiple morphologically similar weeds and crops in a field environment. The finding can also provide a strong reference for the weed identification systems in intelligent weed control equipment.

       

    /

    返回文章
    返回