aitorzip / PyTorch-SRGAN

A modern PyTorch implementation of SRGAN
https://arxiv.org/abs/1609.04802
GNU General Public License v3.0
361 stars 92 forks source link

how to run #5

Closed suke27 closed 6 years ago

suke27 commented 6 years ago

hi, can you tell how to run after train done, seems no test program

suke27 commented 6 years ago

i make test.py by myself, but it seems the image is not right. the out image is garbled. can you share me the pretrain model

aitorzip commented 6 years ago

Hi! I've improved training and added a test script. Take a look at it

suke27 commented 6 years ago

yes, and I check the Generator net, for 1st layer and last layer, in case of K9 why padding set 1, won't it cause pixel lost?

aitorzip commented 6 years ago

True, I replaced by padding 4 a few days ago.