dongheehand / srgan-pytorch Goto Github PK

Can you please provide the code for calculating SSIM

I am trying to use your reo for inference but keep getting this error ,any ideas??
TypeError: unsupported operand type(s) for /: 'str' and 'float'

Hi, thank you for the code & models, @dongheehand !
I used the pretrained model of SRGAN available, and this is what it gave me:
input

output

Now, I wonder on the reasons why the contrast and intensity levels of the output are distributed differently compared to the input. Does anyone have clue how this effect can be avoided?
Thanks!

First of all,thank you for the code，but I have a few questions to ask
1.At the end of the generator you wrote,why there is a Tanh layer.
2.In your gan traing,why g_loss has one more loss L2loss

Much thank you for your reply

Hello. Great works.
I have converted your model to CoreML Apple mobile format.
https://github.com/john-rocky/CoreML-Models

Can you provide the license of your repo?

Thank you.

Loading pre-trained SRGAN.pt model weights, giving it a scale=2. It returns
Error(s) in loading state_dict for Generator: Unexpected key(s) in state_dict: "tail.1.body.0.body.0.weight", "tail.1.body.0.body.0.bias", "tail.1.body.2.weight". scale