CodeFormer AI是一款非常好用的模型开源的视频去码、图片修复软件
处理效果展示
开源项目地址
sczhou/CodeFormer: [NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer (github.com)
下面是本地部署教程(如果你的电脑有很好的gpu,没有的话cpu也可以,会烧)
1.安装环境:
python (建议10左右的版本,太高报错)https://www.python.org/downloads/注意!安装时勾选add .exe to path!否则就得手动加进系统环境里
git https://git-scm.com/
conda https://www.anaconda.com/products/distribution
Pytorch pytorch.org/(根据自己的系统选择,选conda,python,显卡类型)
(英伟达独立显卡,cuda引擎大于10.1。否则只能通过cpu解码)
2.安装其它需求包
cmd或conda powershell 逐条执行下列代码(在管理员权限下进行)
git clone https://github.com/sczhou/CodeFormer
cd CodeFormerpip
install -r requirements.txt -qpip
install -q gradiopython basicsr/setup.py develop
3.【下载预训练模型】(可能要加速):
python scripts/download_pretrained_models.py
facelibpython scripts/download_pretrained_models.py CodeFormer
这时我们就部署完成了!
4.推荐在conda powershell中使用各个功能(不容易出问题),结果将保存在results文件夹中。
面部修复(裁剪和对齐的面部)
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