Plant diseases classification using pre-trained and transfer learning models: A study on rice leaves

Document Type : Research Article


Department of Computer Engineering, Faculty of Engineering, Shahid Chamran University of Ahvaz, Ahvaz, Iran


Ocular detection of pests by phytosanitary specialists, as a very imperative and challenging task, appears to be time consuming, costly, and associated with human error in today’s farming processes. In modern agriculture, diagnostic softwares by artificial intelligence are advised to be used by farmers themselves with little time and cost and with more accuracy. In this paper, two different datasets of rice leaf disease have been used with two transfer learning methods for diagnosing rice leaf disease. The first method uses a CNN-based output of a pre-trained model with an appropriate classifier. In the second method, freezing bottom layers, fine-tuning weights of last layers of the pre-trained network, and adding an appropriate classifier to the model are proposed. For this purpose, seven CNN models have been designed and evaluated. Next, the weights of the best model among these seven proposed models is used to train a second dataset of rice leaves with more disease classes and fewer images. Using these weights, an accuracy of over 96% is reached which is higher than other comparing methods. Furthermore, Grad-CAM, heat map and ROC diagram are used to observe the diagnostic areas of the best model.


Main Subjects