training and evaluating a deep learning model to classify diseases in hydroponic lettuce.
- Model: MobileNetV2 (Transfer Learning)
- Framework: TensorFlow/Keras
- Interface: Gradio Web App
-
Install Dependencies:
pip install -r requirements.txt
-
Dataset Preparation: Place your dataset in a
data/raw/directory. The structure should be:data/ raw/ healthy/ img1.jpg ... downy_mildew/ ... powdery_mildew/ ... -
Training the Model:
python -m src.train
This will save the best model to
models/best_model.kerasand training plots to theoutputs/directory. -
Evaluating the Model:
python -m src.evaluate
This generates a classification report and confusion matrix in the
outputs/directory. -
Running the Web Interface:
python -m src.app
This starts a local web server to test the model on new images with confidence scores.