DOI QR코드

DOI QR Code

Performance Comparison of Base CNN Models in Transfer Learning for Crop Diseases Classification

농작물 질병분류를 위한 전이학습에 사용되는 기초 합성곱신경망 모델간 성능 비교

  • 윤협상 (대구가톨릭대학교 소프트웨어융합학과) ;
  • 정석봉 (경일대학교 철도학부)
  • Received : 2021.07.22
  • Accepted : 2021.08.11
  • Published : 2021.09.30

Abstract

Recently, transfer learning techniques with a base convolutional neural network (CNN) model have widely gained acceptance in early detection and classification of crop diseases to increase agricultural productivity with reducing disease spread. The transfer learning techniques based classifiers generally achieve over 90% of classification accuracy for crop diseases using dataset of crop leaf images (e.g., PlantVillage dataset), but they have ability to classify only the pre-trained diseases. This paper provides with an evaluation scheme on selecting an effective base CNN model for crop disease transfer learning with regard to the accuracy of trained target crops as well as of untrained target crops. First, we present transfer learning models called CDC (crop disease classification) architecture including widely used base (pre-trained) CNN models. We evaluate each performance of seven base CNN models for four untrained crops. The results of performance evaluation show that the DenseNet201 is one of the best base CNN models.

Keywords

References

  1. Agarwal, M., Gupta, S.K., and Biswas, K.K., Development of Efficient CNN model for Tomato Crop Disease Identification, Sustainable Computing: Informatics and Systems, 2020, Vol. 28, 100407. https://doi.org/10.1016/j.suscom.2020.100407
  2. Bhattarai, S., New Plant Diseases Dataset: Image dataset containing different healthy and unhealthy crop leaves, https://www.kaggle.com/vipoooool/new-plant-diseases-dataset.
  3. Ferentinos, K.P., Deep Learning Models for Plant Disease Detection and Diagnosis, Computer Electronics Agriculture, 2018, Vol. 138, pp. 311-318. https://doi.org/10.1016/j.compag.2018.01.009
  4. Goodfellow, I., Bengio, Y., and Courville, A., Deep Learning, MIT Press, 2015, pp.118.
  5. Huang, T., Yang, R., Huang, W., Huang, Y., and Qiao, X., Detecting Sugarcane Borer Diseases Using Support Vector Machine, Information Processing Agriculture, 2018, Vol. 5, No. 1, pp. 74-82. https://doi.org/10.1016/j.inpa.2017.11.001
  6. Jeong, S.B. and Yoon, H.-S., An Efficient Disease Inspection Model for Untrained Crops Using VGG16, Journal of the Korea Society for Simulation, 2020, Vol. 29, No.4, pp. 1-7. https://doi.org/10.9709/JKSS.2020.29.4.001
  7. Jia, D., Wei, D., Richard, S., Li, L.J., Li, K., and Li, F.F., ImageNet: A Large-scale Hierarchical Image Database, Proceedings of IEEE International Conference on Computer Vision and Pattern Recognition, 2009, pp. 248-255.
  8. Khamparia1, A., Saini1, G., Gupta, D., Khanna, A., Tiwari, S., and Albuquerque, V. H. C., Seasonal Crops Disease Prediction and Classification Using Deep Convolutional Encoder Network, Circuits, Systems, and Signal Processing, 2020, Vol. 39, pp. 818- 836. https://doi.org/10.1007/s00034-019-01041-0
  9. Picon, A., Alvarez-Gila, A., Seitz, M., Ortiz- Barredo, A., Echazarra, J., and Johannes, A., Deep Convolutional Neural Networks for Mobile Capture Device-based Crop Disease Classification in the Wild, Computer Electronics Agriculture, 2018, Vol. 138, pp. 200-209. https://doi.org/10.1016/j.compag.2017.04.013
  10. Rangarajan, A.K., Purushothaman, R., and Ramesh, A., Tomato Crop Disease Classification Using Pre-trained Deep Learning Algorithm, Proceedings of the International Conference on Robitcs and Smart Manufacturing, 2018, pp. 1040-1047.
  11. Saleem, M.H., Potgieter, J., and Arif, K.M., Plant Disease Detection and Classification by Deep Learning, Plants, 2019, Vol. 8, pp. 468-490. https://doi.org/10.3390/plants8110468
  12. Sardogan, M., Tuncer, A., and Ozen, Y., Plant Leaf Disease Detection and Classification Based on CNN with LVQ Algorithm, Proceedings of the International Conference on Computer Science and Engineering, 2018, pp. 382-385.
  13. Sladojevic, S., Arsenovic, M., Anderla, A., Culibrk, D., and Stefanovic, D., Deep Neural Networks Based Recognition of Plant Diseases by Leaf Image Classification, Computational Intelligence and Neuroscience, 2016, Vol. 2016, pp.1-11.
  14. Shijie, J., Peiyi, J., Siping, H., Haibo, L. (2017) Automatic detection of tomato disease and pests based on leaf images, Chinese Automation Congress, pp. 3507-3510.
  15. Srivastava, N., Hinton, G., Krizhevsky, A., Sutskever, I., and Salakhutdinov, R., Dropout: a simple way to prevent neural network from overfitting, Journal of Machine Learning Research, 2015, Vol. 15, pp. 1929-1958.