刘斌, 贾润昌, 朱先语, 于聪, 姚卓含, 张海曦, 何东健. 面向移动端的苹果叶部病虫害轻量级识别模型[J]. 农业工程学报, 2022, 38(6): 130-139. DOI: 10.11975/j.issn.1002-6819.2022.06.015
    引用本文: 刘斌, 贾润昌, 朱先语, 于聪, 姚卓含, 张海曦, 何东健. 面向移动端的苹果叶部病虫害轻量级识别模型[J]. 农业工程学报, 2022, 38(6): 130-139. DOI: 10.11975/j.issn.1002-6819.2022.06.015
    Liu Bin, Jia Runchang, Zhu Xianyu, Yu Cong, Yao Zhuohan, Zhang Haixi, He Dongjian. Lightweight identification model for apple leaf diseases and pests based on mobile terminals[J]. Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 2022, 38(6): 130-139. DOI: 10.11975/j.issn.1002-6819.2022.06.015
    Citation: Liu Bin, Jia Runchang, Zhu Xianyu, Yu Cong, Yao Zhuohan, Zhang Haixi, He Dongjian. Lightweight identification model for apple leaf diseases and pests based on mobile terminals[J]. Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 2022, 38(6): 130-139. DOI: 10.11975/j.issn.1002-6819.2022.06.015

    面向移动端的苹果叶部病虫害轻量级识别模型

    Lightweight identification model for apple leaf diseases and pests based on mobile terminals

    • 摘要: 花叶病、斑点落叶病、褐斑病、白粉病、黄蚜、浅叶蛾和红蜘蛛是常见的苹果叶部病虫害,严重影响了苹果的产量和品质。病虫害早期诊断和防治可以有效地控制病害传播,降低损失,保障苹果产业的健康发展。为解决现有轻量级模型无法精准识别早期苹果叶部稀疏小病斑的问题,该研究面向资源受限的移动端设备,提出一种轻量级识别模型ALS-Net(Apple Leaf Net using Channel Shuffle)。在轻量化模型(ShuffleNetV2)的基础上,基于深度可分离卷积和通道混洗构建ALS模块,可降低模型的计算量和参数量。其次,采用知识蒸馏策略训练模型,进一步提高网络精度。试验结果表明,ALS-Net的模型精度可达99.43%,且模型大小仅为1.64 MB。移动端推理延迟为55 ms,能够有效满足实际应用需求,并实现基于移动端的苹果叶部病虫害自动实时监测。

       

      Abstract: Abstract: Apple has been one of the most popular cash crops for the development of the agricultural economy. Seven common diseases have posed a serious threat to the yield and quality of apples, including the Mosaic, Alternaria spot, Brown spot, Powdery mildew, Aphid, Leafminer and Spider mite pests of plant leaves. Early diagnosis and control of diseases and pests can greatly contribute to preventing the spread and reducing losses in the apple industry. However, the current lightweight models cannot accurately identify the sparse small lesions in early apple leaves. In this study, a lightweight recognition model, ALS-Net (Apple Leaf Net using channel shuffle) was proposed for the resource-constrained mobile terminals. The specific procedure was as follows: 1) Firstly, 1 881 images of diseased and healthy apple leaves were collected in Qian County, Shaanxi Province in China. Digital image processing was conducted to enhance the original images, particularly for the generalization and robustness of the model. 2) Secondly, the ALS module was constructed using depth-wise separable convolution and channel shuffle technology. The calculations and parameters of the model were significantly reduced, compared with the traditional convolution network. The channel shuffle technology fully shuffled the information between channels, and then randomly assigned it to each channel. As such, the loss of accuracy was alleviated in group convolution. The Inception structure was introduced into the model for the multi-scale feature extraction. The channel attention was selected to strengthen the disease features in the network, while suppressing the natural background. The Exponential Linear Unit (ELU) was selected as the activation function to accelerate the convergence speed of the model. 3) The knowledge distillation strategy was used to train the model, providing the soft label information that the student model cannot learn on the hard label to realize the transfer of knowledge. The accuracy of the student model was approached or exceeded the accuracy of the teacher model. DenseNet-161 with high accuracy was selected as the teacher model, and the ALS-Net was the student model. A high-performance server was utilized to train the model for the generalization of the model. The experimental results were as follows. 1) The comparative experiment showed that the accuracy of ALS-Net reached 99.43%, which was higher than that of classical CNNs, such as AlexNet and ResNet, and the size of the model was only 1.64 MB, which was lower than that of lightweight CNNs, such as MobileNetV2 and ShuffleNetV2. 2) There were two sets of ablation experiments. The first verified the effects of expanding the convolution kernel to adjust the number of blocks on the model accuracy and parameter number. The second verified the effects of Inception structure, attention module, ELU activation function, and channel shuffle on the accuracy of the model. 3) The training of knowledge distillation strategy significantly improved the recognition accuracy, further accelerating the convergence speed during training. 4) Using PyTorch Mobile, the model was deployed on HUAWEI P40 Pro 5G mobile terminal for real-time inference. The inference delay of the mobile terminal was 55 ms, fully meeting the requirements of practical application. The automatic recognition of apple leaf diseases and pests was realized on the mobile terminal. The finding can provide an insightful idea for the early diagnosis of apple leaf diseases and pests.

       

    /

    返回文章
    返回