Skip to content

Ousoka/cat_dog_classifier

Repository files navigation

Cat vs Dog Image Classifier using Transfer Learning

Project Logo Project Logo

Project Overview

This deep learning project focuses on building a highly accurate image classifier to distinguish between cats and dogs using transfer learning techniques. Our model achieves exceptional performance metrics:

  • Accuracy > 90%
  • Error Rate < 0.8

What makes our model particularly strong is its resilience - it maintains high accuracy even with unclear, blurry, or otherwise challenging images that would typically confuse less robust classifiers.

Key Features

  • Utilizes state-of-the-art transfer learning techniques
  • Robust performance even with imperfect input images
  • Comprehensive model evaluation and visualization
  • Deployed as a user-friendly web application

Technical Approach

We leveraged pre-trained models and fine-tuned them for our specific classification task. The project includes:

  • Data preprocessing and augmentation
  • Model architecture selection
  • Transfer learning implementation
  • Performance optimization
  • Detailed evaluation metrics

Project Team

  • Ousmane KA
  • Albert Sandokh BAKHOUM

Polytech Diamniadio / UAM - Master 2 en Ingénierie des Systèmes d'Informations et de données

Project Artifacts

  1. Jupyter Notebook: Complete implementation with detailed explanations (provided in HTML format)
  2. Web Application: Deployed model accessible online

Live Demo

🔗 https://ok-cat-dog-classifier.streamlit.app/

Try it with your own pet photos!

Model Resilience

Our classifier demonstrates exceptional robustness with the following capabilities:

  • Accurate classification of blurry or low-resolution images
  • Consistent performance with partially obscured animals
  • Reliable predictions for unusual angles or lighting conditions
  • Handles various image artifacts and noise effectively

Getting Started

To explore the project implementation:

  1. Download the provided HTML notebook
  2. Open it in any modern web browser
  3. Follow along with the detailed comments and explanations

Future Enhancements

  • Mobile application integration
  • Real-time classification from video feeds
  • Expanded animal classification capabilities

This project was developed as part of our Master's program in Information and Data Systems Engineering at Polytech Diamniadio / UAM.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published