CNN Transfer Learning for Automatic Image-Based Classification of Crop Disease
As the latest breakthrough in the field of computer vision, deep convolutional neural network(CNN) is very promising for the classification of crop diseases. However, the common limitation applying the algorithm is reliance on a large amount of training data. In some cases, obtaining and labeling a large dataset might be difficult. We solve this problem both from the network size and the training mechanism. In this paper, using 2430 images from the natural environment, which contain 2 crop species and 8 diseases, 6 kinds of CNN with different depths are trained to investigate appropriate structure. In order to address the over-fitting problem caused by our small-scale dataset, we systemically analyze the performances of training from scratch and using transfer learning. In case of transfer learning, we first train PlantVillage dataset to get a pre-trained model, and then retrain our dataset based on this model to adjust parameters. The CNN with 5 convolutional layers achieves an accuracy of 90.84% by using transfer learning. Experimental results demonstrate that the combination of CNN and transfer learning is effective for crop disease images classification with small-scale dataset.
KeywordsCNN Transfer learning Crop disease Image-based classification Over-fitting
The work is partially supported by the National Natural Science Foundation of China (31501223) and Key Laboratory of Agricultural Internet of Things, Ministry of Agriculture, P.R. China.
- 1.Zhang, S.W., Shang, Y.J., Wang, L.: Plant disease recognition based on plant leaf image. J. Anim. Plant Sci. 25(3), 42–45 (2015)Google Scholar
- 3.Kai, S., Liu, Z., Hang, S., Guo, C.: A research of maize disease image recognition of corn based on BP networks. In: Third International Conference on Measuring Technology and Mechatronics Automation, pp. 246–249 (2011)Google Scholar
- 4.Sammany, M., Medhat, T.: Dimensionality reduction using rough set approach for two neural networks-based applications. In: International Conference on Rough Sets and Intelligent Systems Paradigms, pp. 639–647 (2007)Google Scholar
- 5.Wan, J., Wang, D., Hoi, S.C.H., Wu, P., Zhu, J., Zhang, Y., Li, J.: Deep learning for content-based image retrieval:a comprehensive study. In: the ACM International Conference, pp. 157–166 (2014)Google Scholar
- 9.Srdjan, S., Marko, A., Andras, A., Dubravko, C., Darko, S.: Deep neural networks based recognition of plant diseases by leaf image classification. Comput. Intell. Neurosci. 2016(6), 1–11 (2016)Google Scholar
- 11.Amara, J., Bouaziz, B., Algergawy, A.: A deep learning-based approach for banana leaf diseases classification. In: Datenbanksysteme Fr Business, Technologie Und Web, pp. 79–88 (2017)Google Scholar
- 12.Wang, G., Sun, Y., Wang, J.: Automatic image-based plant disease severity estimation using deep learning. Comput. Intell. Neurosci. 2017, 2917536 (2017)Google Scholar
- 14.Deng, J., Dong, W., Socher, R., Li, L.J., Li, K., Li, F.F.: ImageNet: a large-scale hierarchical image database. In: IEEE Conference on Computer Vision and Pattern Recognition, CVPR 2009, pp. 248–255 (2009)Google Scholar
- 15.Massini, M., Fortunato, M., Mancini, S., Tombesi, P.: L2 regularization for learning kernels, vol. 62, no. (4), pp. 109–116 (2012)Google Scholar
- 16.Hinton, G.E., Srivastava, N., Krizhevsky, A., Sutskever, I., Salakhutdinov, R.R.: Improving neural networks by preventing co-adaptation of feature detectors. Comput. Sci. 3(4), 212–223 (2012)Google Scholar
- 17.Kingma, D.P., Ba, J.: Adam: a method for stochastic optimization. Comput. Sci. (2014)Google Scholar