YOLOv11 cat breed detection model using Oxford-IIIT Pet Dataset.
Python · YOLOv11 · FastAPI · PyTorch · Computer VisionCat breed detection system using YOLOv11 model trained on the Oxford-IIIT Pet Dataset to classify 12 different cat breeds.
Features
- Multi-breed Classification: Detects and classifies 12 cat breeds
- YOLOv11: State-of-the-art object detection model
- FastAPI Backend: Real-time prediction API with WebSocket support
- Video Processing: Real-time cat breed detection from video streams
- Kaggle Integration: Public datasets and notebooks on Kaggle
Tech Stack
- Model: YOLOv11 for object detection and classification
- Backend: FastAPI for REST API and WebSocket endpoints
- Training: PyTorch-based YOLO training
- Dataset: Oxford-IIIT Pet Dataset converted to YOLO format
- Package Manager: uv for Python dependency management
Kaggle Resources
- Training notebook, evaluation, and test notebooks available on Kaggle
- Pre-trained weights and datasets published publicly