eugenelet / Keras-Transfer-Learning-of-VGG19-on-Cifar-10-Dataset

PyTorch Transfer Learning of VGG19 on Cifar-10 Dataset

Transfer Learning of VGG19 trained using ImageNet an retrained for Cifar-10 Dataset using PyTorch.

Requirement

Library	Version
Python	^2.7
Keras	^2.0.2
Numpy	^1.12.0
Pickle	*

Usage

Download code:

git clone https://github.com/eugenelet/Keras-Transfer-Learning-of-VGG19-on-Cifar-10-Dataset

cd Keras-Transfer-Learning-of-VGG19-on-Cifar-10-Dataset

Train cnn:

Batch size: 128

Prediction made on per epoch basis.

161 epochs takes about 3h on GTX 1080.

Retrain Model:

python main.py

Tensorboard

tensorboard --logdir=./tensorboard

License

Apache License 2.0

Implementation Details

My Blog