A machine learning pipeline for detecting crop health patterns using Convolutional Neural Networks (CNNs). The system is trained on 10,000+ labeled agricultural images to predict crop conditions with high accuracy, and includes analytics and visualizations of model performance.
- π€ Trained 3 custom CNN architectures using TensorFlow and Keras
- π Achieved 90.07% accuracy on validation data
- π Designed analytics scripts to evaluate 4 key model metrics:
- Accuracy
- Precision
- Recall
- Loss
- π Visualized predictions using Matplotlib with bar graphs and cluster charts
| Purpose | Technology |
|---|---|
| Model Training | TensorFlow, Keras |
| Data Preprocessing | Scikit-learn |
| Data Visualization | Matplotlib |
| Language | Python |
- Dataset: >10,000 images across various crop-health categories
- Preprocessing: Normalization, augmentation, and one-hot encoding
- CNN Variants:
- ResNet-inspired shallow net
- Custom-built 6-layer CNN
- Lightweight MobileNetv2 baseline
- Evaluation: Accuracy calculated via validation split (90.07%)
Using matplotlib, the following charts were generated:
- Bar graphs comparing model performance
- Cluster plots to group prediction categories
- Accuracy/Loss trends over training epochs
train_model.pyβ Model architecture and training loopevaluate.pyβ Script to calculate and compare model metricsvisualize.pyβ Generates performance graphsREADME.mdβ Project documentation
- Integrate early stopping and learning rate schedulers
- Expand dataset with underrepresented crops
- Deploy trained model to a web/mobile interface for farmer use
January β February 2025
Created as a solo AI-agriculture capstone to explore the intersection of deep learning and food security.
Kevin Chifamba
π§ [email protected]
π LinkedIn β’ GitHub