idealo / image-super-resolution

🔎 Super-scale your images and run experiments with Residual Dense and Adversarial Networks.
https://idealo.github.io/image-super-resolution/
Apache License 2.0
4.56k stars 755 forks source link

How to introduce GANS and feature loss in transfer learning? #216

Open Sarah335 opened 2 years ago

Sarah335 commented 2 years ago

I'm looking to fine-tune the noise-cancel weights with MRI brain images and I'm not sure if I should start with only the PSNR loss (e.g. generator loss weight = 1.0, feature extractor and discriminator loss weights = 0), or if I could go straight to using the GANS and feature loss (e.g. generator loss weight = 0, feature extractor loss = 0.0833, discriminator loss = 0.01).

Given the noise-cancel RDN was trained WITH adversarial and feature losses, am I safe to use these from the start?! Curious to hear people's opinions/suggestions!