CodeFormer 安装

感谢阅读

程序功能

图片修复、去水印、黑白还原为彩色等功能

安装

安装anaconda

windows

源码下载
进入cmd

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

linux

git clone https://github.com/sczhou/CodeFormer
cd CodeFormer
pip install -r requirements.txt -q
pip install -q gradio
python basicsr/setup.py develop

下載 facelib 預訓練模型 ，并将其放到weights/facelib下（如果不行，可以用下面的命令）

python scripts/download_pretrained_models.py facelib

下載 CodeFormer 預訓練模型，并将其放到weights/CodeFormer下（如果不行，可以用下面的命令）

python scripts/download_pretrained_models.py CodeFormer

操作方法

面部修复

python inference_codeformer.py -w 0.5--has_aligned --input_path [input folder]

整体图像增强

python inference_codeformer.py -w 0.7--input_path [image folder/image path]

视频增强

python inference_codeformer.py --bg_upsampler realesrgan --face_upsample -w 1.0--input_path

注意事項

1.每次使用前都要先进入cmd，执行cd CodeFormer，进入安装目录后再执行对应命令。
2.执行python开头的命令时，把[input folder]整体替换成待处理的图片所在的文件夹路径。整体图像和视频增强同理，都是把[]替换成路径。
3.结果在result里面
4.w为保真度，取值为0到1