Binary image classifier that distinguishes my cat Ivy from other cats.
Built with ResNet18 transfer learning using PyTorch. Achieved around 98.63% test accuracy trained on ~1400 images of people, landscapes, and other cats.
- Clone the repo
- Set your dataset path in the notebook
- Run all cells
- Go to app.py and replace 'Ivy' with your pet's name!
- Go to https://share.streamlit.io/new and log in with GitHub
- Create the app and have fun! 😁