The objective of this project is to develop a end-to-end image classification model. The model is able to classify the images of 10 dog breeds ( beagle, chihuahua, doberman, french_bulldog, golden_retriever, malamute, pug, saint_bernard, scottish_deerhound, tibetan_mastiff). The dataset is taken from kaggle and pytorch framework is used to train the model. By using the fastapi webframework the model is deployed on heroku for real time uses.
-- Project Link: https://dog-breed-prediction-01.herokuapp.com/
- Pytorch
- Jupyter Notebook
- Numpy, Pandas, Matplotlib
- Fastapi
- HTML, CSS