diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py index d3706f0f648ca5e23fe64e54d913ce64d9fd34a5..173841ab5afcba67af067e43ca9d0d7d7c9b200d 100644 --- a/modules/GSOC/E2_ESRGAN/lib/train.py +++ b/modules/GSOC/E2_ESRGAN/lib/train.py @@ -228,21 +228,26 @@ class Trainer(object): # TDR, not using perceptual loss with CLD OPD # percep_loss = tf.reduce_mean(perceptual_loss(image_hr, fake)) # logging.debug("Calculated Perceptual Loss") - - l1_loss = utils.pixel_loss(image_hr, fake) - logging.debug("Calculated Pixel Loss") - - loss_RaG = ra_gen(image_hr, fake) - logging.debug("Calculated Relativistic" - "Average (RA) Loss for Generator") - - disc_loss = ra_disc(image_hr, fake) - logging.debug("Calculated RA Loss Discriminator") - - - # TDR, we don't have percep_loss - # gen_loss = percep_loss + lambda_ * loss_RaG + eta * l1_loss - gen_loss = lambda_ * loss_RaG + eta * l1_loss + # -------------------------------------------------------------- + + # # substitute standard losses below... + # l1_loss = utils.pixel_loss(image_hr, fake) + # logging.debug("Calculated Pixel Loss") + # + # loss_RaG = ra_gen(image_hr, fake) + # logging.debug("Calculated Relativistic" + # "Average (RA) Loss for Generator") + # + # disc_loss = ra_disc(image_hr, fake) + # logging.debug("Calculated RA Loss Discriminator") + # + # # TDR, we don't have percep_loss + # # gen_loss = percep_loss + lambda_ * loss_RaG + eta * l1_loss + # gen_loss = lambda_ * loss_RaG + eta * l1_loss + # # ----------------------------------------------------------- + + gen_loss = utils.std_loss_G(discriminator, fake) + disc_loss = utils.std_loss_D(discriminator, image_hr, fake) logging.debug("Calculated Generator Loss") disc_metric(disc_loss)