用于苹果叶病害识别的轻量级卷积神经网络
Lightweight-Convolutional Neural Network for Apple Leaf Disease Identification.
作者信息
Fu Lili, Li Shijun, Sun Yu, Mu Ye, Hu Tianli, Gong He
机构信息
College of Information Technology, Jilin Agricultural University, Changchun, China.
College of Electronic and Information Engineering, Wuzhou University, Wuzhou, China.
出版信息
Front Plant Sci. 2022 May 24;13:831219. doi: 10.3389/fpls.2022.831219. eCollection 2022.
As a widely consumed fruit worldwide, it is extremely important to prevent and control disease in apple trees. In this research, we designed convolutional neural networks (CNNs) for five diseases that affect apple tree leaves based on the AlexNet model. First, the coarse-grained features of the disease are extracted in the model using dilated convolution, which helps to maintain a large receptive field while reducing the number of parameters. The parallel convolution module is added to extract leaf disease features at multiple scales. Subsequently, the series 3 × 3 convolutions shortcut connection allows the model to deal with additional nonlinearities. Further, the attention mechanism is added to all aggregated output modules to better fit channel features and reduce the impact of a complex background on the model performance. Finally, the two fully connected layers are replaced by global pooling to reduce the number of model parameters, to ensure that the features are not lost. The final recognition accuracy of the model is 97.36%, and the size of the model is 5.87 MB. In comparison with five other models, our model design is reasonable and has good robustness; further, the results show that the proposed model is lightweight and can identify apple leaf diseases with high accuracy.
作为全球广泛消费的水果,防治苹果树病害极其重要。在本研究中,我们基于AlexNet模型为影响苹果树叶的五种病害设计了卷积神经网络(CNN)。首先,使用空洞卷积在模型中提取病害的粗粒度特征,这有助于在减少参数数量的同时保持较大的感受野。添加并行卷积模块以多尺度提取叶片病害特征。随后,串联的3×3卷积捷径连接使模型能够处理额外的非线性。此外,在所有聚合输出模块中添加注意力机制,以更好地拟合通道特征并减少复杂背景对模型性能的影响。最后,用全局池化替换两个全连接层以减少模型参数数量,确保特征不丢失。该模型的最终识别准确率为97.36%,模型大小为5.87MB。与其他五个模型相比,我们的模型设计合理且具有良好的鲁棒性;此外,结果表明所提出的模型轻量级且能够高精度识别苹果树叶病害。
相似文献
Front Plant Sci. 2022-5-24
Front Plant Sci. 2022-8-22
Front Plant Sci. 2022-11-1
Front Plant Sci. 2023-9-13
Front Plant Sci. 2023-5-9
Front Plant Sci. 2023-8-9
Sensors (Basel). 2020-6-22
引用本文的文献
PeerJ Comput Sci. 2025-1-31
Front Plant Sci. 2023-9-13
本文引用的文献
Front Plant Sci. 2019-7-23
Front Plant Sci. 2016-9-22