Peter-512/deep-transfer-learning

Deep Learning - Transfer Learning

Official hand in to run in a Google colab notebook

Link to Notebook file or already uploaded Notebook here

Quick Start

First, install the dependencies:

pip install -r requirements.txt

Local version

Then, execute all the cells in the notebook preprocess.ipynb in order to do parts of the preprocessing.

Finally, execute all the cells in the notebook deep_learning.ipynb.

Overview of the project

Source data is in images.

Training data is in training_data and testing data is in testing_data each split up in folders which are used as their labels.

Some extra data for the 'breaking the model' part is in breaking_data. Images inside the folder directly are the source images, and the folders inside hold the preprocessed images, again with the directory name serving as the label.