The model was trained on a combined dataset from two sources:
the combined dataset includes the following 7 classes:
- Sigatoka
- Moko
- Insect Pest
- pestalotiopsis
- Bract Mosaic Virus
- cordana
- Healthy
The following CNN architectures were fine-tuned for classification:
- VGG16
- ResNet152V2
- InceptionV3
- MobileNetV2
- LeNet (custom implementation)
The best performing model - VGG16 is saved as model.h5 for inference in the web application.
Follow these steps to set up the project on your local machine.
git clone https://github.com/Manoj632004/Banana_plant_decease_detection
cd banana-leaf-disease-detectionpip install -r requirements.txt- Run the run.ipynb notebook to train your desired CNN model.
- Export the model as model.h5
mkdir staticpython app.py