cydiachen / FSRNET_pytorch

Dear Friend, Having tried several ways, I tried to reproduce the performance of FSRNet by using Pytorch. I am Now transferring to another new project FSRNet, You can download the code and run at your own machine. Feel free to contact me.
73 stars 12 forks source link

FSRNet Pytorch

Dear friends, Thank you for keep tracking in this implementation of FSRNet (CVPR 2018 Oral Paper)

I have been spent the whole summer as an intern in Iluvatar.ai. I have been back to school, so I have time to complete the Project.

I rewrite the Train.py and other model code completely. Now I am uploading pretrained Weights on BaiduNetDisk together with Training Set.

Based on WaveletSRNet, I altered the code by adopting FSRNet network structure.

Prerequisites

Train

Change the option in Train.py to set the dataset's directory. I am using CelebAHQ-MASK as the training set. The GroundTruth is generated by zllrunning/face-parsing.PyTorch(https://github.com/zllrunning/face-parsing.PyTorch) with pretrained model.

Dataset Link: https://pan.baidu.com/s/1HEECUyKI5GOSrd7NPlm-ow 密码:z2ud

Test

ON GOING :| PYTHON AND NOTEBOOK WILL BE PROVIDED. Pretrained Weights:链接:https://pan.baidu.com/s/1ZkgABGefsMjO6XhhvlBzRA 密码:libl

Result

Citation

If you find FSRNet useful in your research, please consider citing (* indicates equal contributions):

@inproceedings{CT-FSRNet-2018,
  title={FSRNet: End-to-End Learning Face Super-Resolution with Facial Priors},
  author={Chen, Yu* and Tai, Ying* and Liu, Xiaoming and Shen, Chunhua and Yang, Jian },
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  year={2018}
}