/tesorflow-vgg

vgg implemention with tesorflow.

Primary LanguagePython

VGG-16 implementation with tensorflow & pytorch

Data set is CIFAR-10, click here to download CIFAR-10 python version.
Extract cifar-10-python.tar.gz and put the folder to the root of the project.
  • vgg-16.py : vgg 16 implementation with tensorflow.
  • vgg-16-pytorch.py : vgg 16 implementation with pytorch.
  • simple-network.py : a simple two layer conventional neural network.
  • others.py : they are not working right now.

Fell free to fork and create pull request.