johnhany
/ mnist-convnet-lenet-visualize.py
Created
September 25, 2017 14:01
A demo for MNIST LeNet model for TensorFlow, including exporting model and visualization.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import os | |
| import tensorflow as tf | |
| LOGDIR = 'visualize/' | |
| from tensorflow.examples.tutorials.mnist import input_data | |
| mnist = input_data.read_data_sets("F:\\Datasets\\AorB\\mnist\\", one_hot=True) | |
| def conv_layer(input, size_in, size_out, name="conv"): | |
| with tf.name_scope(name): |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #include <iostream> | |
| #include <opencv2/opencv.hpp> | |
| using namespace std; | |
| using namespace cv; | |
| #define PIXELS_IN_IMAGE 28*28 | |
| #define ENABLE_TRAIN 1 |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #include <iostream> | |
| #include <opencv2/opencv.hpp> | |
| using namespace std; | |
| using namespace cv; | |
| #define PIXELS_IN_IMAGE 28*28 | |
| #define ENABLE_TRAIN 1 |