Skip to content

A machine learning project for classifying images of cats and dogs

Notifications You must be signed in to change notification settings

eskayML/cat-and-dogs-classification

Repository files navigation

cat-and-dogs-classification

View Live on Spaces

A machine learning project for classifying images of cats and dogs

Curated a dataset containing 25000 images of cats and dogs

STEPS I TOOK BUILDING THE APPLICATION:


- I built a model using convnets architecture and applied techniques like data augmentation and transfer learning
- I then fitted the model to a larger portion of the data (training set). Note that I used a smaller portion(about 20%) for testing/validation
- Checked the performance on the test data
- I built an interface for the model using streamlit - Live deployment to hugging face spaces

HOW TO USE THE APP LOCALLY

$ git clone https://github.com/eskayML/cat-and-dogs-classification
$ cd cat-and-dogs-classification
$ pip install -r requirements.txt

Install the dependencies using the command above and then run

$ streamlit run main.py

About

A machine learning project for classifying images of cats and dogs

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published