彭红星, 徐慧明, 刘华鼐. 基于改进ShuffleNet V2的轻量化农作物害虫识别模型[J]. 农业工程学报, 2022, 38(11): 161-170. DOI: 10.11975/j.issn.1002-6819.2022.11.018
    引用本文: 彭红星, 徐慧明, 刘华鼐. 基于改进ShuffleNet V2的轻量化农作物害虫识别模型[J]. 农业工程学报, 2022, 38(11): 161-170. DOI: 10.11975/j.issn.1002-6819.2022.11.018
    Peng Hongxing, Xu Huiming, Liu Huanai. Lightweight agricultural crops pest identification model using improved ShuffleNet V2[J]. Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 2022, 38(11): 161-170. DOI: 10.11975/j.issn.1002-6819.2022.11.018
    Citation: Peng Hongxing, Xu Huiming, Liu Huanai. Lightweight agricultural crops pest identification model using improved ShuffleNet V2[J]. Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 2022, 38(11): 161-170. DOI: 10.11975/j.issn.1002-6819.2022.11.018

    基于改进ShuffleNet V2的轻量化农作物害虫识别模型

    Lightweight agricultural crops pest identification model using improved ShuffleNet V2

    • 摘要: 及时准确地识别害虫是有效防治的重要前提。针对现有基于卷积神经网络的害虫识别模型实时性差、识别率低、结构复杂不易部署等问题,提出基于改进ShuffleNet V2的农作物害虫识别模型。首先,在ShuffleNet V2中引入多尺度特征融合模块LMFF(Lightweight Multi-scale Feature Fusion),加强模型对不同尺度害虫的特征提取能力;其次,在ECA(Efficient Channel Attention)注意力机制中增加并行路径,并通过可学习参数自适应更新不同路径的权重,提出AECA(Adaptive and Efficient Channel Attention)注意力机制,将AECA注意力机制嵌入到ShuffleNet V2中,提高模型的跨通道交互能力;然后,使用SiLU(Sigmoid Weighted Liner Unit)替换ReLU激活函数,增强模型的泛化能力;最后,通过调整输出通道数和核心模块的堆叠次数重新设计ShuffleNet V2的整体架构,降低模型的计算量和参数量,从而提出轻量化的农作物害虫识别模型SNPF(ShuffleNet for Pest Field)。试验结果表明,SNPF模型在自建害虫数据集上的平均识别准确率和F1分数为79.49%和78.54%,较改进前分别提高了4.00个百分点和3.09个百分点,而参数量和浮点运算量为3.74 M和0.48 G,较改进前分别下降了30.60%和18.60%。SNPF模型对单张害虫图像的平均推理时间为11.9 ms,与ResNet 50、GoogLeNet、EfficientNet B1等模型相比,SNPF模型的识别精度更高,并且识别时间分别减少了57.04%、50.21%和40.50%。该研究提出的SNPF模型能够较好地识别农作物害虫、并且具有识别速度快和轻量化的特点,可以为农作物害虫的防治提供帮助。

       

      Abstract: Pests and diseases have posed a serious threat to the growth and storage of crops in recent years, leading to major grain losses worldwide. Various chemical substances (such as insecticides) have been widely used to control pests, where some adverse effects have occurred in the agricultural ecosystem. Timely and accurate identification of pests can be one of the most important steps for effective control. However, traditional pest monitoring relies mainly on experts or technicians to manually identify pests, indicating the subjective, labor-intensive, costly, and difficult translation on a large scale. Alternatively, the convolutional neural networks (CNN) in the field of deep learning can be widely expected as one of the best ways to promote conventional identification. However, the current CNNs suffer from low real-time performance, identification accuracy, and complex structures for easy use during pest identification. In this study, an identification model was proposed for the pests of lightweight crops using an improved ShuffleNet V2 CNN. Firstly, the Lightweight Multi-scale Feature Fusion (LMFF) module was designed using depth-separable and pointwise convolution. Three branches were contained in the LMFF module, including the 3×3 depth-separable convolution, 1×1 pointwise convolution, and residual structure. The basic unit structure was introduced into the ShuffleNet V2 to obtain the receptive fields of different scales, particularly for better feature extraction for the pests. Secondly, the parallel paths were added to the Efficient Channel Attention (ECA) mechanism, in order to adaptively update the weights of different paths by learnable parameters. The Adaptive and Efficient Channel Attention (AECA) attention mechanism was then obtained to introduce into the 1×1 point-wise convolution of the ShuffleNet V2 basic unit and 1×1 convolution of each branch of the sub-sampling unit. As such, the cross-channel interaction ability of the model was improved significantly. A Sigmoid Weighted Liner Unit (SiLU) activation function was also established to avoid neuron necrosis in the ReLU activation function. A better performance was achieved in the smooth and non-monotonic SiLU on deep networks, compared with the ReLU. Therefore, the SiLU activation function replaced the ReLu one in the basic and downsampling unit of ShuffleNet V2 for a better overall performance of the model. Finally, the overall architecture of the model was redesigned to reduce the parameters and floating point operations per second (FLOPs). A ShuffleNet for Pest Field (SNPF) model was then constructed for pest identification. Specifically, the number of stacks in the core module was reduced by more than half. The numbers of output channels in the first and last convolutional layers were adjusted from 24 to 32, and from 2048 to 1024, respectively. The experimental results showed that the average identification accuracy and the F1 Score of SNPF on the self-built pest dataset were 79.49% and 78.54%, which were 4.00 percentage points and 3.09 percentage points higher than before, respectively. The parameters and FLOPs of the SNPF model were about 3.74 M and 0.48 G, which were 30.60% and 18.60% lower than the original, respectively. The average identification time of SNPF was 11.9 ms for a single pest image, which was reduced by 57.04%, 50.21%, and 40.50%, respectively, compared with the ResNet 50, GoogLeNet, and EfficientNet B1 models, respectively. The SNPF model can be expected to rapidly and accurately identify crop pests for better pest control. The finding can also greatly contribute to the automatic identification and efficient control of pests during lightweight crop production.

       

    /

    返回文章
    返回