Simple VGG implementation for image classification using tensorflow and python. Easy to use for beginners like me.
Python 2.7.12 Tensorflow 1.3.0
Put your data according to the labels (1,2,3,...,n) into different folder. Training: ./example/train/1, ./example/train/2 ,... , ./example/train/n Testing: ./example/test/1, ./example/test/2 ,... , ./example/test/n
Or one dataset using cross validation ./easy/1, ./easy/2,..., ./easy/n
To alter between one dataset cross validation and training & testing set, comment or uncomment the corresponding codes.
This model performs better with relatively large amount of training data. The dataset in this repository is for demonstration, the mere amount of which is insufficient for acquiring desirable performance.
For more information (in Chinese), please visit my blog