CodeFormer: Blind face restoration with transformer codebook lookup
Transformer-based face restoration using vector-quantized codebook lookup.
Learn more about CodeFormer
from basicsr.archs.codeformer_arch import CodeFormer\nnet = CodeFormer().cuda()\noutput = net(input_image, w=0.5, fidelity_weight=0.5)
Codebook-based restoration
Uses vector quantized codebook lookup to map degraded facial regions to learned high-quality representations, rather than direct pixel-level regression. This approach constrains outputs to realistic face distributions learned during training.
Blind restoration capability
Operates without explicit knowledge of degradation type or severity, handling mixed degradation scenarios including compression artifacts, noise, blur, and missing regions. The model generalizes across different degradation patterns without task-specific fine-tuning.
Multi-task framework
Supports face restoration, inpainting, and colorization through a unified architecture with task-specific checkpoints. Includes video processing support for temporal consistency and integration with multiple face detection backends including dlib and RetinaFace.
from basicsr.archs.codeformer_arch import CodeFormer
import torch
net = CodeFormer().cuda()
net.eval()
with torch.no_grad():
restored_face = net(input_face_tensor, w=0.5, fidelity_weight=0.5)
result = restored_face[0].cpu().clamp(0, 1)CodeFormer V0.1.0 Release
- –This release is mainly for storing pre-trained models, etc.
Top in AI & ML
Related Repositories
Discover similar tools and frameworks used by developers
ComfyUI
Visual graph-based diffusion model workflow builder.
ControlNet
Dual-branch architecture for conditional diffusion model control.
llama.cpp
Quantized LLM inference with hardware-accelerated CPU/GPU backends.
X Recommendation Algorithm
Open source implementation of X's recommendation algorithm for timeline and notification ranking.
docling
Fast document parser for RAG and AI workflows.