Tensorflow-srcnn
A tensorflow based super-resolution convolutional neural network model. Distributed learning supported.
Files
srcnn.py: Single computer running file.
d_srcnn.py: Distributed training file.
tensor_srcnn.py: SRCNN model file.
predict.py: Put the test image into the network and get the gray(light) image as result.
generate_train.m: Caffe model original training data generation file.
generate_test.m: Caffe model original training data generation file.
generate_train_im.m: Training data generation file for ImageNet images with control of the output size.
Details
Executing command follows the guide here: https://www.tensorflow.org/versions/r0.9/how_tos/distributed/index.html#putting-it-all-together-example-trainer-program. Also, tensorflow should be installed.
The input data file is a single hdf5 file called train.h5 by default. Changes is needed when large training data is used.
For testing, one can compute psnr of the output of predict.py and the Y channel of the original picture.