/GPEN

Primary LanguagePython

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

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
  • Download RetinaFace model and our pre-trained model (not our best model due to commercial issues) and put them into weights/.

    RetinaFace-R50 | GPEN-512 | GPEN-1024-Color

  • Restore face images:

python face_enhancement.py
  • Colorize aligned faces:
python face_colorization.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.