BlueAmulet / BasicSR

Fork of Basic Super-Resolution codes for development. Includes ESRGAN, SFT-GAN for training and testing.
Apache License 2.0
18 stars 1 forks source link

Implement HR Downscale #8

Closed joeyballentine closed 3 years ago

joeyballentine commented 4 years ago

Adds 3 new options to the training config: hr_downscale, hr_downscale_types, hr_downscale_amount.

Will downscale HR images by the value set in the config, or will choose randomly from an array. Will also downscale LR images if provided, unless the resulting HRs are actually the correct scale to the LRs.

This allows you to have one high-quality dataset but cover multiple scales (either in separate models or a single model), as well as use a 4x dataset for a 2x model with the same LRs.