Publication Type:

Conference Paper


2020 5th International Conference on Communication and Electronics Systems (ICCES) (2020)


agriculture, classification, CNNs, convolutional neural nets, Convolutional neural networks, Convolutional Neural Networks (CNN), Crops, data augmentation, Deep learning, Deep Neural Network, learning (artificial intelligence), Plant Diseases, PyTorch, ResNet 50, ResNet 50 model, ResNet model, tomato crops, tomato leaf disease detection model, Tomato Leaf Diseases, tomato leaves, Transfer learning


This project briefs the detection of diseases present in a tomato leaf using Convolutional Neural Networks (CNNs) which is a class under a deep neural network. As an initial step, the dataset is segregated before the detection of tomato leaves. The concept of transfer learning is used where a pre-trained model (ResNet-50) is imported and adjusted according to our classification problem. To increase the quality of the ResNet model and to enhance the result as close to the actual prevailing disease, data augmentation has been implemented. Taking all these into consideration, a tomato leaf disease detection model has been developed using PyTorch that uses deep - CNNs. Finally, the testing dataset is processed for validation based on the learned parameters from the ResNet 50 model. Six most prevailing diseases in tomato crops have been taken for classification. Data augmentation has been introduced to increase the data set to 4 times the actual data and the model has shown an accuracy of 97%.

Cite this Research Publication

N. K. E., M., K., P., P., R., A., and S. Veni, “Tomato Leaf Disease Detection using Convolutional Neural Network with Data Augmentation”, in 2020 5th International Conference on Communication and Electronics Systems (ICCES), 2020.