Tomato Disease Classification using Fine-Tuned Convolutional Neural Network


  • Haseeb Younis Department of Computer Science, University College Cork, Cork, Ireland
  • Muhammad Asad Arshed Department of Software Engineering, University of Management & Technology, Lahore, Pakistan
  • Fawad ul Hassan Superior University, Lahore Campus, Pakistan
  • Maryam Khurshid Comsats University Islamabad, Lahore Campus, Pakistan
  • Hadia Ghassan Department of Computer Science, Minhaj University, Lahore, Pakistan


Classification of Disease Tomato Plants, Mobile-Net, Plant Disease Classification, Transfer Learning


Tomatoes have enhanced vitamins that are necessary for mental and physical health. We use tomatoes in our daily life. The global agricultural industry is dominated by vegetables. Farmers typically suffer a significant loss when tomato plants are affected by multiple diseases. Diagnosis of tomato diseases at an early stage can help address this deficit. It is difficult to classify the attacking disease due to its range of manifestations. We can use deep learning models to identify diseased plants at an initial stage and take appropriate measures to minimize loss through early detection. For the initial diagnosis and classification of diseased plants, an effective deep learning model has been proposed in this paper. Our deep learning-based pre-trained model has been tuned twofold using a specific dataset. The dataset includes tomato plant images that show diseased and healthy tomato plants. In our classification, we intend to label each plant with the name of the disease or healthy that is afflicting it. With 98.93% accuracy, we were able to achieve astounding results using the transfer learning method on this dataset of tomato plants. Based on our understanding, this model appears to be lighter than other advanced models with such considerable results and which employ ten classes of tomatoes. This deep learning application is usable in reality to detect plant diseases.

Full Text


S. Savary, L. Willocquet, S. J. Pethybridge, P. Esker, N. McRoberts, and A. Nelson, "The global burden of pathogens and pests on major food crops," Nat. Ecol. Evol., vol. 3, no. 3, pp. 430–439, Mar. 2019.

S. Mansoor et al., "Evidence for the association of a bipartite geminivirus with tomato leaf curl disease in Pakistan," Plant Disease, vol. 81, no. 8. Plant Dis, p. 958, 1997.

A. Raza et al., "First Report of Tomato Chlorosis Virus Infecting Tomato in Pakistan," Plant Dis., vol. 104, no. 7, pp. 2036–2036, Jul. 2020.

J. G. A. Barbedo, "A review on the main challenges in automatic plant disease identification based on visible range images," Biosystems Engineering, vol. 144. Academic Press, pp. 52–60, 01-Apr-2016.

A. García-Floriano, Á. Ferreira-Santiago, O. Camacho-Nieto, and C. Yáñez-Márquez, “A machine learning approach to medical image classification: Detecting age-related macular degeneration in fundus images,” Comput. Electr. Eng., vol. 75, pp. 218–229, May 2019.

W. Zhao and S. Du, "Spectral-Spatial Feature Extraction for Hyperspectral Image Classification: A Dimension Reduction and Deep Learning Approach," IEEE Trans. Geosci. Remote Sens., vol. 54, no. 8, pp. 4544–4554, Aug. 2016.

M. Z. Khan et al., "A Realistic Image Generation of Face from Text Description Using the Fully Trained Generative Adversarial Networks," IEEE Access, vol. 9, pp. 1250–1260, 2021.

G. Khan, S. Jabeen, M. Z. Khan, M. U. G. Khan, and R. Iqbal, "Blockchain-enabled deep semantic video-to-video summarization for IoT devices," Comput. Electr. Eng., vol. 81, p. 106524, Jan. 2020.

M. S. Mustafa, Z. Husin, W. K. Tan, M. F. Mavi, and R. S. M. Farook, "Development of an automated hybrid intelligent system for herbs plant classification and early herbs plant disease detection," Neural Comput. Appl., vol. 32, no. 15, pp. 11419–11441, Aug. 2020.

M. S. Mustafa, Z. Husin, W. K. Tan, M. F. Mavi, and R. S. M. Farook, "Development of an automated hybrid intelligent system for herbs plant classification and early herbs plant disease detection," Neural Comput. Appl., vol. 32, no. 15, pp. 11419–11441, Aug. 2020.

J. G. A. Barbedo, "Impact of dataset size and variety on the effectiveness of deep learning and transfer learning for plant disease classification," Comput. Electron. Agric., vol. 153, pp. 46–53, Oct. 2018.

C. DeChant et al., "Automated identification of northern leaf blight-infected maize plants from field imagery using deep learning," Phytopathology, vol. 107, no. 11, pp. 1426–1432, Nov. 2017.

H. Younis, M. H. Bhatti, and M. Azeem, "Classification of skin cancer dermoscopy images using transfer learning," in 15th International Conference on Emerging Technologies, ICET 2019, 2019.

A. Fuentes, S. Yoon, S. Kim, and D. Park, "A Robust Deep-Learning-Based Detector for Real-Time Tomato Plant Diseases and Pests Recognition," Sensors, vol. 17, no. 9, p. 2022, Sep. 2017.

K. KC, Z. Yin, M. Wu, and Z. Wu, "Depthwise separable convolution architectures for plant disease classification," Comput. Electron. Agric., vol. 165, p. 104948, Oct. 2019.

D. Oppenheim and G. Shani, "Potato Disease Classification Using Convolution Neural Networks," 2017.

H. Qassim, A. Verma, and D. Feinzimer, "Compressed residual-VGG16 CNN model for big data places image recognition," in 2018 IEEE 8th Annual Computing and Communication Workshop and Conference, CCWC 2018, 2018, vol. 2018-January, pp. 169–175.

Y. Lu, S. Yi, N. Zeng, Y. Liu, and Y. Zhang, "Identification of rice diseases using deep convolutional neural networks," Neurocomputing, vol. 267, pp. 378–384, Dec. 2017.

M. Z. Alom et al., "The History Began from AlexNet: A Comprehensive Survey on Deep Learning Approaches," arXiv, Mar. 2018.

A. Sinha and R. Singh Shekhawat, "A novel image classification technique for spot and blight diseases in plant leaves," Imaging Sci. J., vol. 68, no. 4, pp. 225–239, May 2020.

J. Chen, J. Chen, D. Zhang, Y. A. Nanehkaran, and Y. Sun, "A cognitive vision method for the detection of plant disease images," Mach. Vis. Appl., vol. 32, no. 1, p. 3, Feb. 2021.

A. G. Howard et al., “MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications,” arXiv, Apr. 2017.




How to Cite

Younis, H., Muhammad Asad Arshed, Fawad ul Hassan, Maryam Khurshid, & Hadia Ghassan. (2022). Tomato Disease Classification using Fine-Tuned Convolutional Neural Network. International Journal of Innovations in Science & Technology, 4(1), 123–134. Retrieved from