thatbrguy / VGG19-Transfer-Learn

TensorFlow implementation of VGG19 with Transfer Learning

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

VGG19-Transfer-Learn

TensorFlow implementation of VGG19 with Transfer Learning. The model is based on this implementation.

This repository is used as a teaching aid, for demonstraing the effectiveness of data augmentation, in this medium blog.

Requirements:

  • Python 3.5
  • TensorFlow 1.4
  • Keras (Only for converting labels to one-hot, you can replace that with numpy)
  • Jupyter

Steps:

  • Fork/Clone this repository.
  • Download VGG19.npy from here, place it in this repository.
  • Download the example data from here.
  • Run the jupyter notebook.

About

TensorFlow implementation of VGG19 with Transfer Learning


Languages

Language:Jupyter Notebook 92.5%Language:Python 7.5%