Skip to content

Commit

Permalink
Merge pull request #867 from Varunshiyam/fixes-863
Browse files Browse the repository at this point in the history
Potato Disease Classification with CNN
  • Loading branch information
UppuluriKalyani authored Nov 10, 2024
2 parents 70042cd + 7996e15 commit af97f29
Show file tree
Hide file tree
Showing 2 changed files with 1,065 additions and 0 deletions.
33 changes: 33 additions & 0 deletions Prediction Models/Potato_disease_classification/Readme.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
# Potato Disease Classification with CNN

## Project Overview

This project implements a Convolutional Neural Network (CNN) model to classify diseases in potato plants from leaf images. The model is trained using image data preprocessing techniques, including data augmentation and normalization, to enhance performance. This tool aids in identifying various potato diseases efficiently and accurately.

## Problem Statement

Identifying potato diseases accurately is critical in agriculture to prevent crop loss and ensure high yield. Traditional methods can be slow and error-prone, particularly when scaled. This project aims to develop a machine learning model to automatically detect and classify potato leaf diseases, thereby supporting better disease management and decision-making in agriculture.

## Project Structure

- `data/`: Contains the dataset for training and testing.
- `notebooks/`: Jupyter notebooks documenting model development and testing.
- `src/`: Source code for data preprocessing, model building, training, and evaluation.
- `README.md`: Project documentation.

## Model Architecture

The model uses a CNN with:
- Convolutional and pooling layers for feature extraction
- Dense layers for classification
- Softmax activation for multi-class output

Data augmentation techniques like random flips and rotations are applied to increase robustness.

## Dependencies

- TensorFlow
- Keras
- NumPy
- Matplotlib

Loading

0 comments on commit af97f29

Please sign in to comment.