CodeFormer 免费开源项目 GitHub :【链接】、【网盘下载】
如果你 电脑没安装conda,那么请先下载安装【官方链接】
安装命令:
git clone https://github.com/sczhou/CodeFormer
cd CodeFormer
pip install -r requirements.txt
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
面部修复(裁剪和对齐的面部)
# For 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文件夹中。