Denisganga / the_plant_doctor

This project aims to develop a machine learning model to detect and classify plant diseases using the Plant Village dataset.

Plant Disease Classification using PyTorch

This repository contains the implementation of a plant disease classification model using PyTorch. The model is based on a fine-tuned ResNet-18 architecture for image classification tasks of the plantvillage dataset.

Table of Contents

• Introduction
• Dataset
• Installation
• Usage
• Training
• Evaluation
• Results
• Model Saving
• Customization
• Contributing
• License

Introduction

The goal of this project is to classify plant diseases based on images. The model is implemented in PyTorch and utilizes data augmentation and a pre-trained ResNet-18 model for improved accuracy.

Dataset

The dataset used for this project is available here. It consists of images of plants affected by various diseases. The dataset is organized into different classes based on the type of disease.The dataset is available for download at kaggle, dataset name Plantvillage dataset

Installation

1. Clone the repository:

   git clone https://github.com/denisganga/the_plant_doctor.git
   cd the_plant_doctor

2. Install the required dependencies:

   pip install -r requirements.txt

Usage

1. Ensure the dataset is downloaded and organized in the specified structure.

2. Modify the config.py file to set the dataset path and other configurations.

3. Run the training script:

   python train.py

4. Evaluate the model:

   python evaluate.py

Training

The model is trained using a fine-tuned ResNet-18 architecture. Data augmentation techniques are applied to enhance the model's robustness. The training script (train.py) contains the main training loop.

Evaluation

The model is evaluated on a separate test set to measure its accuracy and performance. The evaluation script (evaluate.py) computes the accuracy and validation loss.

Results

The trained model achieves an accuracy of approximately 51% on the test set. Further details can be found in the Results section of the code.

Model Saving

The trained model is saved using torch.save and can be loaded for future use. The saved model file (fine_tuned_resnet18.pth) contains the model's state dictionary.

Customization

Feel free to customize the code, experiment with hyperparameters, or try different pre-trained models for improved performance. You can also contribute by adding new features or enhancing existing ones.

Contributing

Contributions are welcome, i humbly request for a collaboration on this notebook for an improvement on the accuracy! Feel free to open issues, submit pull requests, or provide feedback.

License

This project is licensed under the MIT_license.