pytorch-SRResNet icon indicating copy to clipboard operation
pytorch-SRResNet copied to clipboard

LeakyReLU?

Open BrunoVox opened this issue 5 years ago • 0 comments

Regarding your SRResNet implementation (the generator), I was wondering why you used a LeakyReLU instead of a PReLU with a trainable "a". Reading the paper and seing your implementation raised this question, because the two are related, but a bit different. It seems to me that the author meant a trainable PReLU in the original implementation.

Also, why did you use InstanceNorm2d instead of BatchNorm2d? Were you feeding the network with single images instead of a 16 sized batch as the paper specifies?

My last question would be what dataset you used to train your implementation? I've seen you used data augmentation, but I didn't find info on the used dataset.

I'm studying the paper and planning on running it on Imagenet, as the author did, expecting to get similar results. I also will definetly take a look on your evaluation code implementation, the Matlab tip, which increases PSNR values, seems valuable.

My implementation is a bit different than yours and it's getting 29,75dB PSNR on Set5. To get these results I trained it with a sampled 5500 images from Imagenet and I'm hoping to get an improvement when I train it with the 350k images they mention in the paper.

BrunoVox avatar Apr 10 '19 23:04 BrunoVox