Skip to content

zdyshine/BSRGAN_implementation

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

What is the repository about?

This code is the unofficial implementation of this work. Specifically, I coded the pipeline of the whole degradation model according to the configuration given in the paper. Note that some details are different from the original paper (For the sake of simplicity, I've removed the processed camera sensor noise, but this part may be key to the generalization of the network). I chose HiFaceGAN as the model for restoring the degraded images. I used the code degrade.py to generate 10000 degraded face images in advance. After training with two V100 on hifacegan for one day, the reconstruction effect on the real images I found is much better than the general bicubic degradation method.

Update

I added the code of the processed camera sensor noise, thanks for this work, but I have not tested the effectiveness of this method. Note that my implementation of this method is slightly different from the BSRGAN author's paper.

Comparison

The degraded images below are all real images from social networks.

Images layout is LR | mixed degradation SR

Images layout is LR | bicubic degradation SR | mixed degradation SR

What else to do?

Compared with the original paper, the size of current training set of my method is relatively small and the training time is relatively short, though it has achieved much better results than the bicubic degradation method. In some real images, there are still some noises that can't be removed, which may be because I didn't use the processed camera sensor noise. In some test images, it is also found that the reconstruction effect of the model for the face area is acceptable, but the reconstruction effect of the background and text is relatively poor, which may be caused by the lack of diversity of the data set itself.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%