CodeFormer 安装

感谢阅读

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

程序功能

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

安装

安装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

你可能感兴趣的:(机器学习,python,人工智能,深度学习)