This project aims to classify diseases in plant leaves using deep learning and computer vision techniques. It includes four main parts:
- Data Analysis: Analyze and visualize the dataset to understand class distributions. The dataset consists of images of plant leaves with various diseases, organized into subdirectories by plant type and disease category.
- Data Augmentation: Balance dataset classes and improve model generalization using various augmentation techniques.
- Image Transformation: Apply image transformation methods for feature extraction and model learning.
- Classification: Develop and train convolutional neural network (CNN) models to accurately classify diseases in plant leaves.
To analyze the dataset and generate pie charts and bar charts for each plant type.
python3 Distribution.py /path/to/dataset
To create augmented images using flip, rotate, crop, distortion , brightness and saturation techniques.
python3 Augmentation.py /path/to/image
To apply image transformation methods such as crop, skew, and rotation for feature extraction:
python3 Transformation.py /path/to/image
To train CNN model classify diseases in plant leaves.
python3 train.py /path/to/dataset
To predict which disease the image represents :
python3 predict.py /path/to/image
- Python 3.x
- TensorFlow
- Keras
- NumPy
- OpenCV
- Matplotlib