Unofficial implementation of FSRCNN using PyTorch.
- Run
data_aug.m
to augment training set. - Run
generate_train.m
andgenerate_test.m
to generate h5 files for data loader. (Note that you may need to change the path in the matlab code for different image source) - Run
main.py
.
- python >= 3.4
- pytorch >= 0.4
- torchvision
- tensorflow
Chao Dong, Chen Change Loy, Xiaoou Tang. Accelerating the Super-Resolution Convolutional Neural Network, in Proceedings of European Conference on Computer Vision (ECCV), 2016