Skip to content

Image classifier for dog breeds using EfficientNetB3 pre-trained model and transfer learning. The model achieves 94.8% training accuracy and 90% validation accuracy across 120 dog breeds. Features automatic pipeline for data loading, augmentation, training and evaluation using TensorFlow/Keras.

Notifications You must be signed in to change notification settings

SamH135/dog-breed-image-classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 

Repository files navigation

Dataset: https://www.kaggle.com/c/dog-breed-identification

Instructions:
You can find our notebook named "DogBreed_Classifier.ipynb" which contains all the code for the model. In order to run the code, run each cell block in sequential order (in Google Colab). All the necessary files/data will be automatically downloaded off a public GitHub repository.

Output file:
The file "Image Classifier Output" contains all the output statistics/plots pertaining to the project. You can also refer to specific cell blocks within the notebook file for further analysis.

NOTE: Running the notebook file requires a lot of Google Colab tokens (if you're running it on Colab). Your run may be interrupted if you use the base version of Google Colab.

About

Image classifier for dog breeds using EfficientNetB3 pre-trained model and transfer learning. The model achieves 94.8% training accuracy and 90% validation accuracy across 120 dog breeds. Features automatic pipeline for data loading, augmentation, training and evaluation using TensorFlow/Keras.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published