感谢阅读

  • 程序功能
  • 安装
    • windows
    • linux
  • 操作方法
    • 面部修复
    • 整体图像增强
    • 视频增强
    • 注意事項

程序功能

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

安装

安装anaconda

windows

源码下载
进入cmd

cd CodeFormerpip install -r requirements.txt -qpip install -q gradiopython basicsr/setup.py develop

linux

git clone https://github.com/sczhou/CodeFormercd CodeFormerpip install -r requirements.txt -qpip install -q gradiopython 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