codeformer-pip

PyTorch implementation of CodeFormer


Keywords
pytorch, codeformer, face, face-swap, face-swapping, face-swap-pytorch, face-swapping-pytorch
License
Other
Install
pip install codeformer-pip==0.0.4

Documentation

Towards Robust Blind Face Restoration with Codebook Lookup Transformer

pypi version HuggingFace Spaces

This repo is a PyTorch implementation of the paper CodeFormer.

Installation

pip install codeformer-pip

Usage

from codeformer.app import inference_app

inference_app(
  image="test.jpg",
  background_enhance=True,
  face_upsample=True,
  upscale=2,
  codeformer_fidelity=0.5,
)

Citation

@inproceedings{zhou2022codeformer,
    author = {Zhou, Shangchen and Chan, Kelvin C.K. and Li, Chongyi and Loy, Chen Change},
    title = {Towards Robust Blind Face Restoration with Codebook Lookup TransFormer},
    booktitle = {NeurIPS},
    year = {2022}
}

License

This project is licensed under NTU S-Lab License 1.0. Redistribution and use should follow this license.