wind222 / deep_cnn

Deep convolution neural network project with TensorFlow

Deep CNN

The classic deep convolution neural network project with TensorFlow. You can learn and run CNN easily.

• Train/test/inference images included.
• Visualize images with matplotlib.
• Support checkpoint and tensorboard.
• Easy to extend more convolution layers.

Train

./pokemon_classifer.py --epoch_number 100

Inference

./pokemon_classifer.py --mode inference --image ./data/inference/Pikachu.png