Skip to content

A deep learning project for identifying diseases in rice leaves using transfer learning with the ResNet152V2 model incorporating channel attention mechanism

License

Notifications You must be signed in to change notification settings

cse-kiet/PCSE24-38

Repository files navigation

Rice Leaf Disease Identification

Overview

Rice Leaf Disease Identification is a deep learning project aimed at detecting and classifying diseases in rice leaves using computer vision techniques. The project utilizes transfer learning with the ResNet152V2 model, incorporating an attention layer to focus on important parts of the image. The dataset used for training the model contains approximately 2600 images belonging to six different classes.

Project Video

Watch the video here

Dataset

The dataset used for training the model is sourced from Kaggle and contains images of rice leaves affected by different diseases. You can access the dataset here.

Model Training

The deep learning model is trained using transfer learning techniques with the ResNet152V2 architecture. Transfer learning allows leveraging pre-trained models on large datasets and fine-tuning them for specific tasks. In this project, the ResNet152V2 model is fine-tuned on the rice leaf disease dataset to classify different types of diseases.

Classes

The model is trained to classify rice leaf images into the following disease classes:

  • Healthy
  • Brown spot
  • Leaf blast
  • Bacterial leaf blight
  • Leaf scald
  • Narrow brown spot

Deployment

The trained model is deployed using Flask, a lightweight Python web framework. It accepts images of rice leaves as input and identifies the disease present in the leaf.

Usage (To execute code)

To use the deployed model:

  1. Clone the repository.
git clone https://github.com/cse-kiet/PCSE24-38.git
  1. Install the required dependencies specified in the requirements.txt file.
pip install -r requirement.txt
  1. Run the Flask application.
flask --app app run
  1. Upload an image of a rice leaf to the application.
  2. The application will predict the disease present in the leaf and display the result.

Contributors

About

A deep learning project for identifying diseases in rice leaves using transfer learning with the ResNet152V2 model incorporating channel attention mechanism

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •