Cat vs Dog Classifier
A deep learning web application for classifying images of cats and dogs using MobileNetV2 transfer learning with a FastAPI backend.
The Challenge
Accurately classifying images in real-time is challenging without specialized knowledge. This project democratizes access to computer vision by providing a simple web interface where anyone can upload an image and get an instant, AI-powered classification.
Core Features
🐱
Image Classification
Upload an image to classify it as a cat or dog.
🔄
Model Retraining
Upload your own dataset to retrain the model.
⚡
Real-time Interaction
Instant feedback loop for testing model predictions.
Technology Stack
Python
The primary language for AI and backend development.
FastAPI
High-performance framework for serving the ML model.
TensorFlow/Keras
Used for loading and running the MobileNetV2 model.
Jinja2
Serves the HTML interface directly from FastAPI.
Key Advantages
- State-of-the-art Transfer Learning (90%+ accuracy)
- Lightweight and fast inference
- Simple and intuitive user interface
- Deployable on any cloud platform with Python support