CodeFormer
CodeFormer免费开源项目
https://github.com/sczhou/CodeFormer
安装环境
python git conda
安装命令
git clone https://github.com/sczhou/CodeFormer
cd CodeFormer
pip install -r requirements.txt -q
pip install -q gradio
python basicsr/setup.py develop
下载预训练模型
1.从 [ Google Drive |下载 facelib 预训练模型 OneDrive ] 到weights/facelib文件夹。您可以手动下载预训练模型或通过运行以下命令下载。
python scripts/download_pretrained_models.py facelib
2.从 [ Google Drive |下载 CodeFormer 预训练模型 OneDrive ] 到weights/CodeFormer文件夹。您可以手动下载预训练模型或通过运行以下命令下载。
python scripts/download_pretrained_models.py CodeFormer
面部修复(裁剪和对齐的面部)
#or cropped and aligned faces
python inference_codeformer.py -w 0.5 --has_aligned --input_path [input folder]
整体图像增强
# For whole image
# Add '--bg_upsampler realesrgan' to enhance the background regions with Real-ESRGAN
# Add '--face_upsample' to further upsample restorated face with Real-ESRGAN
python inference_codeformer.py -w 0.7 --input_path [image folder/image path]
视频增强
# For video clips
python inference_codeformer.py --bg_upsampler realesrgan --face_upsample -w 1.0 --input_path
保真度权重w位于 [0, 1] 中。通常,较小的w往往会产生较高质量的结果,而较大的w会产生较高保真度的结果。
结果将保存在results文件夹中。
GFPGAN
https://huggingface.co/spaces/akhaliq/GFPGAN
自建开源项目
https://github.com/sczhou/CodeFormer