hhb072 / WaveletSRNet

A pytorch implementation of Paper "Wavelet-srnet: A wavelet-based cnn for multi-scale face super resolution"
MIT License
240 stars 67 forks source link

How could I get the dataset? #6

Open RemyEE opened 5 years ago

RemyEE commented 5 years ago

I am trying to running the code. It shows an error: No such file or directory: 'path/celeba/train.list' which mean I have to get the CelebA dataset first. But how should I get the dataset and how should I matching the directories?

dogegg1994 commented 5 years ago

I have same problem

RemyEE commented 5 years ago

@dogegg1994 You can check my fork, here: https://github.com/RemyEE/WaveletSRNet Now I have WaveletSRNet working on python 3 on Windows. Check the train.list and test.list for details of training, testing set. By the way, after get face centered data set, I used scipy bicubic interpolation to generated the training and testing set. Good luck.

dogegg1994 commented 5 years ago

thank you

liujianzhao6328057 commented 5 years ago

@dogegg1994 You can check my fork, here: https://github.com/RemyEE/WaveletSRNet Now I have WaveletSRNet working on python 3 on Windows. Check the train.list and test.list for details of training, testing set. By the way, after get face centered data set, I used scipy bicubic interpolation to generated the training and testing set. Good luck.

Could you please upload the pretrained file since I want to test the model quickly?Appreciate your help.

RemyEE commented 5 years ago

@liujianzhao6328057 I can not get the result show in the paper. Even using original source code. I can only get PSNR 24.7db after 500 epochs. I will share the pretrained model after I uploaded it.

hwpengTristin commented 5 years ago

@liujianzhao6328057 I can not get the result show in the paper. Even using original source code. I can only get PSNR 24.7db after 500 epochs. I will share the pretrained model after I uploaded it.

Would you like to share your pre-trained model?