ESRGAN icon indicating copy to clipboard operation
ESRGAN copied to clipboard

Training ESRGAN from scratch without VGG feature extractor

Open donCorleone97 opened this issue 5 years ago • 1 comments

Hi, inspired from your model, I have written a similar ESRGAN model with spectral_normalization and a few other tweaks. Although mostly, the underlying model architecture is same. I am also using VGG19_54 for feature extraction. I'm trying to train the model to super resolve images which VGG19 has never been trained on.

My generator is not producing the highest quality images, there are a few distortions, although the perception of image is excellent. I believe the fault lies in the discriminator not being able to discriminate between fake and real images. Will you please suggest something I can do to mitigate it?

I was thinking if I have to remove VGG_19 part, I can extend the discriminator to be a proper robust CNN OR I can train a VGG19 model without pretrained weights on my own dataset and then use it to extract features. Will that work?

donCorleone97 avatar Dec 16 '19 06:12 donCorleone97

@donCorleone97 did you figure out how to solve this?

kpagels avatar May 27 '20 12:05 kpagels