Model training scale: 2 ESRGAN

[wolfam0108]

Hello! Can I train an ESRGAN model at x2 scale? This requires a trained model for x2. Where can you get such a model?

[xinntao]

@wolfam0108 Sorry that we have not provided models for X2. You can modify the network structure and train it by yourself.

[wolfam0108]

Thanks for the answer. Is there a description of how to change the structure of the ESRGAN network? Or at least in what direction to dig?

[xinntao]

1. A reference: https://github.com/xinntao/BasicSR/blob/master/basicsr/models/archs/srresnet_arch.py
2. Since ESRGAN is already very large and slow for X4, it is better to first downsample the feature maps, and then upsample for 4 times, when applied on X2 scaling factor. (This is called UNet structure. )

[UESTCrookieLI]

hello wolfam0108 did you complete the x2ESRGAN work? I am going to run ESRGAN when scale=2, but i got some problems.