The CLIP-MLOps-Pipeline is a Machine Learning Operations (MLOps) pipeline designed to train, evaluate, and deploy CLIP-based models for image-text processing. It includes modular components for data processing, model training, experiment tracking with MLflow, API deployment with FlaskAPI, and CI/CD automation using Jenkins and Docker.
Modular Design: Organized structure for models, data, training, and testing.
MLflow Tracking: Tracks model metrics and versioning.
Dockerized Deployment: API and training environments are containerized.
CI/CD Integration: Automated pipeline with Jenkins for testing and deployment.
FlaskAPI for Model Serving: Provides an efficient API for inference.
π¦ CLIP-MLOps-Pipeline
βββ π src/
β βββ π models/
β β βββ image_encoder.py
β β βββ text_encoder.py
β β βββ transformer_encoder.py
β β βββ attn_head.py
β β βββ pos_embeds.py
β β βββ model.py
β βββ π data/
β β βββ dataset.py
β β βββ parameters.json
β βββ π training/
β β βββ train.py
β β βββ test.py
β βββ π api/
β β βββ main.py
β β βββ Dockerfile
β β βββ requirements.txt
β βββ π tests/
β β βββ test_model.py
β β βββ test_api.py
βββ π mlflow/
β βββ tracking.py
βββ π docker/
β βββ Dockerfile
βββ π ci-cd/
β βββ Jenkinsfile
βββ README.md
βββ requirements.txt
βββ .gitignore