Computer Science ›› 2022, Vol. 49 ›› Issue (11A): 211200009-6.doi: 10.11896/jsjkx.211200009

• Image Processing & Multimedia Technology • Previous Articles     Next Articles

Identification Method of Maize Disease Based on Transfer Learning and Model Compression

DENG Peng-fei, GUAN Zheng, WANG Yu-yang, WANG Xue   

  1. School of Information,Yunnan University,Kunming 650500,China
  • Online:2022-11-10 Published:2022-11-21
  • About author:DENG Peng-fei,born in 1991,postgra-duate.His main research interests include deep learning and image proces-sing.
    GUAN Zheng,born in 1982,Ph.D,associate professor,is a member of China Computer Federation.Her main research interests include deep learning,intelligent traffic control and so on.
  • Supported by:
    National Natural Science Foundation of China(61761045).

Abstract: Traditional image recognition methods have low accuracy in recognizing maize disease images,and convolutional neural networks have a good effect on image recognition.However,the network model has a large amount of calculation and a large amount of parameters,making it difficult to rely on mobile devices with limited computing power to popularize and use in small sample applications.Therefore,this paper aims to improve the accuracy of maize disease images,reduce network parameters and model size,and proposes a convolutional neural network model that combines migration learning and model compression for maize disease recognition.In order to improve the generalization of the model,the data set is enhanced and the structure of convolutional neural network based on transfer learning is constructed.In this paper,the migration recognition of common maize disease images is carried out by using the improved VGG16-Inception network model pre-trained on ImageNet through transfer learning.Experimental results show that the average recognition accuracy of maize disease images using transfer learning is 93.38% on ImageNet data set.After the migration,the model is compressed by channel pruning and knowledge distillation,and the compressed model is used to recognize the maize disease image by transfer learning.Experimental results show that the average recognition accuracy of corn disease image after compression reaches 92.40%,the accuracy is reduced by 0.98%,the model size is compressed from 73.90 MB to 9.45 MB,and the number of parameters is reduced by 87.80%.The proposed method can ensure the recognition accuracy in small sample scenarios and further realize the model lightweight.

Key words: Convolutional neural network, Maize disease image, Transfer learning, Image identification, Model compression

CLC Number: 

  • TP391
[1]MENG Y,CHEN G F,LU J,et el.Simulink platform real-time diagnosis in the video image of corn diseases[J].Journal of Jilin Agricultural University,2017,39(4):483-487.
[2]LV G Z,CHEN J,BAI J K,at el.Present situation and control measures of corn diseases in China[J].Plant Protection,1997,23(4):20-21.
[3]MILLER S A,BEED F D,HARMON C L.Plant disease diagnostic capabilities and networks[J].Annual Review of Phytopathology,2009,47(1):15-38.
[4]LI Q Q,GUO M K,GUO C,et al.Occurrence dynamics of maize diseases in Gansu Province[J].Plant Protection,2014,40(3):161-164.
[5]KUSSUL N,LAVRENIUK M,SKAKUN S,et al.Deep learning classification of land cover and crop typs using remote sensing data[J/OL].IEEE Geoscience and Remote Sensing Letters,2017,PP(99):1-5.https://xueshu.baidu.com/usercenter/paper/show?paperid=b525470daa7f9623c2f138d8cca55d17&site=xueshu_se.
[6]KRIZHEVSKY A,SUTSKEVER I,HINTON G.ImageNetclassifica-tion with deep convolutional neural networks[J].Advances in Neural Information Processing Systems,2012,25(2):1097-1105.
[7]SIMONYAN K,ZISSERMAN A.Very deep convolutional networks for large-scale image recognition[J].arXiv:1409.1556,2014.
[8]SZEGEDY C,LIU W,JIA Y,et al.Going deeper with convolutions[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2015:1-9.
[9]HE K,ZHANG S R,SUN J.Deep residual learning forimage recognition[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2015:770-778.
[10]YANG Y,ZHANG Y L,MIAO W,et al.Accurate identification and location of corn rhizome based on Faster R-CNN[J].Transactions of the Chinese Society for Agricultural Machinery,2018,49(10):46-53.
[11]GAO Y,GUO J L,LI X,et al.Instance-level segmentationmethod for group pig images based on deep learning[J].Transactions of the Chinese Society for Agricultural Machinery,2019,50(4):179-187.
[12]BI S,GAO F,CHEN J W,et al.Detection method of citrus based on deep convolution neural network[J].Transacions of the Chinese Society for Agricultural Machinery,2019,50(5):181-186.
[13]LIU Y B,LEI B,CAO Y,et al.Recognition of maize disease based on deep convolutional Neural network[J].Chinese Agricultural Science Bulletin,2018,34(36):15-9-164.
[14]YANG M Y,ZHANG Y G,LIU T.Small sample recognition of maize disease based on convolutional neural network[J].Chinese Journal of Eco-Agriculture(Chinese and English),2020,28(12):1924-1931.
[15]XU Y,ZHAO B,ZHAI Y,et al.Maize Diseases Identification Method Based on Multi-Scale Convolutional Global Pooling Neural Network[J].IEEE Access,2021,9:27959-27970.
[16]HAN S,MAO H Z,DALLY W J.Deep compression:com-pressing deep neural networks with pruning,trained quantization and huffman coding[J].Fiber,2015,56(4):3-7.
[17]HINTON G,VINYALS O,DEAN J.Distilling the knowledge in a neural network[J].arXiv:1503.02531,2014.
[1] ZHOU Le-yuan, ZHANG Jian-hua, YUAN Tian-tian, CHEN Sheng-yong. Sequence-to-Sequence Chinese Continuous Sign Language Recognition and Translation with Multi- layer Attention Mechanism Fusion [J]. Computer Science, 2022, 49(9): 155-161.
[2] FANG Yi-qiu, ZHANG Zhen-kun, GE Jun-wei. Cross-domain Recommendation Algorithm Based on Self-attention Mechanism and Transfer Learning [J]. Computer Science, 2022, 49(8): 70-77.
[3] CHEN Yong-quan, JIANG Ying. Analysis Method of APP User Behavior Based on Convolutional Neural Network [J]. Computer Science, 2022, 49(8): 78-85.
[4] ZHU Cheng-zhang, HUANG Jia-er, XIAO Ya-long, WANG Han, ZOU Bei-ji. Deep Hash Retrieval Algorithm for Medical Images Based on Attention Mechanism [J]. Computer Science, 2022, 49(8): 113-119.
[5] DAI Zhao-xia, LI Jin-xin, ZHANG Xiang-dong, XU Xu, MEI Lin, ZHANG Liang. Super-resolution Reconstruction of MRI Based on DNGAN [J]. Computer Science, 2022, 49(7): 113-119.
[6] LIU Yue-hong, NIU Shao-hua, SHEN Xian-hao. Virtual Reality Video Intraframe Prediction Coding Based on Convolutional Neural Network [J]. Computer Science, 2022, 49(7): 127-131.
[7] XU Ming-ke, ZHANG Fan. Head Fusion:A Method to Improve Accuracy and Robustness of Speech Emotion Recognition [J]. Computer Science, 2022, 49(7): 132-141.
[8] WU Zi-bin, YAN Qiao. Projected Gradient Descent Algorithm with Momentum [J]. Computer Science, 2022, 49(6A): 178-183.
[9] WANG Jun-feng, LIU Fan, YANG Sai, LYU Tan-yue, CHEN Zhi-yu, XU Feng. Dam Crack Detection Based on Multi-source Transfer Learning [J]. Computer Science, 2022, 49(6A): 319-324.
[10] CHU Yu-chun, GONG Hang, Wang Xue-fang, LIU Pei-shun. Study on Knowledge Distillation of Target Detection Algorithm Based on YOLOv4 [J]. Computer Science, 2022, 49(6A): 337-344.
[11] YANG Yue, FENG Tao, LIANG Hong, YANG Yang. Image Arbitrary Style Transfer via Criss-cross Attention [J]. Computer Science, 2022, 49(6A): 345-352.
[12] YANG Jian-nan, ZHANG Fan. Classification Method for Small Crops Combining Dual Attention Mechanisms and Hierarchical Network Structure [J]. Computer Science, 2022, 49(6A): 353-357.
[13] ZHANG Jia-hao, LIU Feng, QI Jia-yin. Lightweight Micro-expression Recognition Architecture Based on Bottleneck Transformer [J]. Computer Science, 2022, 49(6A): 370-377.
[14] WANG Jian-ming, CHEN Xiang-yu, YANG Zi-zhong, SHI Chen-yang, ZHANG Yu-hang, QIAN Zheng-kun. Influence of Different Data Augmentation Methods on Model Recognition Accuracy [J]. Computer Science, 2022, 49(6A): 418-423.
[15] SUN Jie-qi, LI Ya-feng, ZHANG Wen-bo, LIU Peng-hui. Dual-field Feature Fusion Deep Convolutional Neural Network Based on Discrete Wavelet Transformation [J]. Computer Science, 2022, 49(6A): 434-440.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!