Jun 11, 2026
Image Classification Pipeline
End-to-end CNN pipeline for multi-class image classification with training dashboards and a deployed inference API.
PyTorchFastAPIDocker
Overview
A production-style image classification system built with PyTorch.
Highlights
- Custom CNN + transfer learning (ResNet-50)
- Data augmentation and experiment tracking
- FastAPI inference endpoint with Docker deployment
What I learned
Handling class imbalance and serving models at low latency.