JiahuiYu / wdsr_ntire2018

Code of our winning entry to NTIRE super-resolution challenge, CVPR 2018
http://www.vision.ee.ethz.ch/ntire18/
598 stars 123 forks source link

The difference between nn.PixelShuffle and nn.ConvTranspose2d ? #18

Closed YongboLiang closed 5 years ago

YongboLiang commented 5 years ago

Hi, I have a doubt about nn.PixelShuffle and nn.ConvTranspose2d. In the upsampling layer, some use nn.PixelShuffle while others use nn.ConvTranspose2d. The number of filters belong to nn.PixelShuffle is scale^2 times that of number of filters belong to nn.ConvTranspose2d. However, I don't find any authors describing the difference between the two methods in their paper.

Have you thought about this and experimented? Will these two methods have an impact on the training results?

Best~

JiahuiYu commented 5 years ago

Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network

https://arxiv.org/abs/1609.05158