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