MLOps Image Classification Project

This project demonstrates how to build an image classification model and organize it using MLOps best practices.

The project uses TensorFlow to train a deep learning model for classifying images (cats vs dogs). The goal is not only to build a model but also to structure the project in a production-ready MLOps format.

Project Structure

configs/ – configuration files for experiments
data/ – dataset storage
models/ – trained models
notebooks/ – experimentation notebooks
reports/ – evaluation results
src/ – training and pipeline code

Running Training

Run the training pipeline:

python run_training.py

Technologies Used

TensorFlow
Python
YAML configuration
GitHub version control