VGG_tensorflow icon indicating copy to clipboard operation
VGG_tensorflow copied to clipboard

VGGNet implemented by tensorflow

VGG_tensorflow

this code implement VGG16 network by this paper.

graph

Training

  • load pre-trained parameters from here: https://mega.nz/#!YU1FWJrA!O1ywiCS2IiOlUCtCpI6HTJOMrneN-Qdv3ywQP5poecM
  • the pre-trained parameters were trained on ImageNet DataSet, 1000 clasess.
  • Remove the final FC layer and add one 10 nodes FC layer to apply for Cifar10 DataSet https://www.cs.toronto.edu/~kriz/cifar.html

first training

use the pre_trained convolution layer parameters, and train the FC layer parameters

Result

second training

Do not use the pre-trained parameter, the network train all parameters

Result