codeformer-pip 0.0.1

PyTorch implementation of CodeFormer

Towards Robust Blind Face Restoration with Codebook Lookup Transformer

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.