Greetings!
This is the repository for my Capstone Project 2 for the Springboard Data Science Career Track. The project/report/deck is entitled "Who’s a good dog?
An exploration and assessment of using Convolutional Neural Networks to classify dog breeds from images".
In this repository, you will find the following folders and files:
-
Main Repo
-
-
Directory Setup:Verify compressed dataset files, create train test directories and crop images according to annotation. Pickling and packaging images into single Numpy array.
-
dog_breed_EDA: Exploring a small selection of images, validating data, sizing up the dataset.
-
First iteration of building CNN from scratch (simple_CNN.ipynb): Initial iterations of CNN models from scratch (before Directory Set up was performed).
-
CNN from Scratch using Pre-processed image data (simple_CNN_preprocessed_images.ipynb): Final iteration of CNN models from Scratch which became the baseline reference for pre-trained models.
-
Using Transfer Learning (vgg_model.ipynb): Using VGG16 model architecture and pre-trained weights. Includes visualizations and predictions on test and real-world data.
-
Deprecated:(simple_CNN_smaller_class_subset.ipynb) Tried using CNN from scratch using flow_from_directory on smaller subset of data.
-
=======