Skip to content

This project explores 4 pre-trained machine learning algorithms and tests their accuracies on the Stanford Dog dataset. There is also the model_deployment.py script that you can download and classify other dog images.

Notifications You must be signed in to change notification settings

Wyatt-Ritchie/dog-breed-classifier

Repository files navigation

CSCI4050U Final Project

https://www.youtube.com/watch?v=UmLHT3xk5PM

Dog Breed Classification using Transfer Learning

Our group did a comparative study of different neural networks. We compared ResNet and VGGNet by classifying dog breeds and plotting the accuracies of each network. This repository contains code for the Jupyter Notebook, the saved ResNet models, and records from each training epoch. We used the Stanford Dogs Dataset in our training. Also, there is code for identifying dog breeds given user image input (via the my_dogs directory).

Contributors

Our team consists of the following members, with their contributions listed:

About

This project explores 4 pre-trained machine learning algorithms and tests their accuracies on the Stanford Dog dataset. There is also the model_deployment.py script that you can download and classify other dog images.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •