CodeFormer

1. 下载

   git clone https://github.com/sczhou/CodeFormer​

2. 根据系统配置安装pyTorch

   https://pytorch.org/get-started/locally/​

3. 安装对应组件

   cd CodeFormer
   pip install -r requirements.txt -q
   pip install -q gradio
   python basicsr/setup.py develop​

4. 下载模型

   python scripts/download_pretrained_models.py facelib
   python scripts/download_pretrained_models.py CodeFormer​

5. 处理图片

   //面部修复
   python inference_codeformer.py -w 0.5 --has_aligned --input_path 图片目录
   
   //整体图像增强
   python inference_codeformer.py -w 0.1 --input_path 图片目录 
   
   //视频增强
   python inference_codeformer.py --bg_upsampler realesrgan --face_upsample -w 1.0 --input_path 视频目录
   
   //面部着色 
   python inference_colorization.py --input_path 图片目录
   
   //面部修复
   python inference_inpainting.py --input_path​ 图片目录

注意：本文归作者所有，未经作者允许，不得转载