Leaffliction

Computer vision: Image classification by disease recognition on leaves

🚀 Getting Started

1. Clone the repository

git clone https://github.com/adiaz-uf/Leaffliction.git
cd Leaffliction

2. Create environment and install dependencies

make

Part 1: Analysis of the Data Set

python src/Part1/Distribution.py Apple/

Part 2: Data augmentation

Displays different data augmentation techniques applied to the image or directory of images passed as argument.

make augment-one

make augment-apple

make augment-apple-train

make augment-grape

make augment-grape-train

Part 3: Image transformations

make transform-one

make transform-apple

make transform-grape

Part 4: Classification

Model predictions

This program takes as arguments a path to an image from our learnings, displays the original image and the transformed image, and gives the type of disease specified in the leaf.

Train uses the images passed as argument to train the model, and creates leaf_disease_model.h5 and class_indices.json files.

You must train the model first and then run the predictions.

make train

make predict

Team work 💪

This project was a team effort. You can checkout the team members here:

• Alejandro Díaz Ufano Pérez
  • Github
  • LinkedIn
  • 42 intra
• Alejandro Aparicio
  • Github
  • LinkedIn
  • 42 intra