Skip to content

Implement lenet and vgg19 by tensorflow with dataset mnist using tfrecord

Notifications You must be signed in to change notification settings

ppplinday/TFrecord-Lenet-Vgg19-tensorflow

Repository files navigation

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

python3 tfrecord.py

For training

python3 train.py tenet

python3 train.py vgg19

For eval python3 eval.py tenet

python3 eavl.py vgg19

You also can change the config in config.py

About

Implement lenet and vgg19 by tensorflow with dataset mnist using tfrecord

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages