Enhancing visual quality of satellite imagery with ESRGAN.
Upscale low-resolution (LR) inputs to high-resolution (HR) outputs.
Combination of pixel loss, perceptual loss, and adversarial loss.
from ununennium.models.gan import ESRGAN
from ununennium.models.gan.losses import PerceptualLoss
generator = ESRGAN(scale_factor=4)
perceptual_criterion = PerceptualLoss()
# In training loop
fake_hr = generator(lr_img)
loss_p = perceptual_criterion(fake_hr, real_hr)