Skip to content

Year 2, Principles of Artificial Intelligence ISB42403, Final Project, TensorFlow-Keras CNN Model Training, Machine Learning

Notifications You must be signed in to change notification settings

puureya2/machine-learning-tensorflow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

2 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

🌾 Crops Sorter Model

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.


🧠 Core Features

  • πŸ€– 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

πŸ› οΈ Tech Stack

Purpose Technology
Model Training TensorFlow, Keras
Data Preprocessing Scikit-learn
Data Visualization Matplotlib
Language Python

πŸ§ͺ Training Overview

  • 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%)

πŸ“Š Visualizations

Using matplotlib, the following charts were generated:

  • Bar graphs comparing model performance
  • Cluster plots to group prediction categories
  • Accuracy/Loss trends over training epochs

πŸ“ Files

  • train_model.py β€” Model architecture and training loop
  • evaluate.py β€” Script to calculate and compare model metrics
  • visualize.py β€” Generates performance graphs
  • README.md β€” Project documentation

🧠 Future Improvements

  • Integrate early stopping and learning rate schedulers
  • Expand dataset with underrepresented crops
  • Deploy trained model to a web/mobile interface for farmer use

πŸ“… Timeline

January – February 2025
Created as a solo AI-agriculture capstone to explore the intersection of deep learning and food security.


πŸ“¬ Contact

Kevin Chifamba
πŸ“§ [email protected]
πŸ”— LinkedIn β€’ GitHub


About

Year 2, Principles of Artificial Intelligence ISB42403, Final Project, TensorFlow-Keras CNN Model Training, Machine Learning

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages