项目作者: ppplinday

项目描述 :
Implement lenet and vgg19 by tensorflow with dataset mnist using tfrecord
高级语言: Python
项目地址: git://github.com/ppplinday/TFrecord-Lenet-Vgg19-tensorflow.git


VGG19-Lenet-tensorflow

This is the tenet model and vgg19 model for cifar10, acc for lenet is about 84% and acc for vgg19 is about 86%.

My code shows how to write tfrecord, data argument, train, eval and model.

For building tfrecord

  1. python3 tfrecord.py

For training

  1. python3 train.py tenet
  2. python3 train.py vgg19

For eval
python3 eval.py tenet

  1. python3 eavl.py vgg19

You also can change the config in config.py