DEV Community

Vansh Goel
Vansh Goel

Posted on

Cat Dog Classification CNN Project

πŸš€ Deployed: Cat vs Dog Image Classification App on Vercel! 🐢🐱

Excited to share my latest Machine Learning projectβ€”a complete end-to-end Cat vs Dog Image Classification web application!

Live Demo: https://cat-and-dog-classification.vercel.app

Features -->
πŸ“€ Upload an image of a cat or dog
🧠 Deep learning prediction using a custom PyTorch CNN
πŸ“Š Displays prediction confidence
πŸ“ˆ Model evaluation metrics:
Accuracy
Precision
Recall
F1-Score
πŸ”² Confusion Matrix visualization
🎯 Responsive React frontend
⚑ FastAPI backend for inference
πŸ› οΈ Tech Stack
Frontend: React.js
Backend: FastAPI
Machine Learning: PyTorch
Deployment: Vercel (Frontend)

This project helped me understand the complete ML deployment workflowβ€”from training a CNN model to building an API and integrating it with a modern frontend.

I'd love to hear your feedback! Feel free to try it out and let me know what you think.

MachineLearning #DeepLearning #PyTorch #FastAPI #ReactJS #AI #ComputerVision #WebDevelopment #Vercel #Python #MLProjects #OpenToWork

Top comments (0)