ESRGAN icon indicating copy to clipboard operation
ESRGAN copied to clipboard

Training a new model with different scaling factor

Open chhsieh0309 opened this issue 5 years ago • 3 comments

Hi there,

If I want to train a new ESRGAN model, a 2.5x upsampling/downsampling factor, for example, how do I get the corresponding pre-trained PSNR model?

chhsieh0309 avatar May 13 '19 07:05 chhsieh0309

I think you can use the weights of pre-trained X4 ESRGAN to initialize the X2.5 model, except the upsampling layer.

xinntao avatar May 13 '19 17:05 xinntao

@chhsieh0309 have you successfully trained the model with 2.5x scale factor? I was able to train with scale factor of integer values (e.g. 2x, 4x), but not sure how to train a model with non-integer scale factor (e.g. 1.5x, 2.5x).

zhangcl avatar Dec 23 '19 19:12 zhangcl

@chhsieh0309 have you successfully trained the model with 2.5x scale factor? I was able to train with scale factor of integer values (e.g. 2x, 4x), but not sure how to train a model with non-integer scale factor (e.g. 1.5x, 2.5x).

why not just use 4x and then downsample to 2.5

jorjiang avatar Nov 09 '21 10:11 jorjiang