Skip to content
main
Switch branches/tags
Code

Latest commit

Files

Permalink
Failed to load latest commit information.

GAN Prior Embedded Network for Blind Face Restoration in the Wild

Paper | Supplementary | Demo

Tao Yang1, Peiran Ren1, Xuansong Xie1, Lei Zhang1,2
1DAMO Academy, Alibaba Group, Hangzhou, China
2Department of Computing, The Hong Kong Polytechnic University, Hong Kong, China

Face Restoration

Face Colorization

Face Inpainting

Conditional Image Synthesis (Seg2Face)

News

(2021-07-06) The training code will be released soon. Stay tuned.

Usage

python pytorch cuda driver gcc

  • Clone this repository:
git clone https://github.com/yangxy/GPEN.git
cd GPEN
python face_enhancement.py
  • Colorize faces:
python face_colorization.py
  • Complete faces:
python face_inpainting.py
  • Synthesize faces:
python segmentation2face.py

Main idea

Citation

If our work is useful for your research, please consider citing:

@inproceedings{Yang2021GPEN,
    title={GAN Prior Embedded Network for Blind Face Restoration in the Wild},
    author={Tao Yang, Peiran Ren, Xuansong Xie, and Lei Zhang},
    booktitle={IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
    year={2021}
}

License

© Alibaba, 2021. For academic and non-commercial use only.

Acknowledgments

We borrow some codes from Pytorch_Retinaface and stylegan2-pytorch.

Contact

If you have any questions or suggestions about this paper, feel free to reach me at yangtao9009@gmail.com.

About

No description, website, or topics provided.

Resources

Releases

No releases published

Packages

No packages published