github地址:https://github.com/sczhou/CodeFormer#face-color-enhancement-and-restoration
CodeFormer 是一款AI 视频、图片修复软件。
git clone https://github.com/sczhou/CodeFormer
安装完成后可在当前目录看到:CodeFormer
cd CodeFormer
Conda 是一个开源的软件包管理系统和环境管理系统,用于安装和管理多个不同版本的软件包和它们的依赖关系
brew install anaconda
如果没有brew,还需要安装brew
/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
安装完成可以通过如下命令查看:
brew --version
conda安装完成之后需要再bash中配置PATH
export PATH=/opt/homebrew/anaconda3/bin:$PATH
3.3 anaconda env
conda create -n codeformer python=3.8 -y
conda activate codeformer
执行如下三个命令
pip3 install -r requirements.txt
python basicsr/setup.py develop
conda install -c conda-forge dlib
# facelib and dlib 训练模型
python scripts/download_pretrained_models.py facelib
python scripts/download_pretrained_models.py dlib
# CodeFormer训练模型
python scripts/download_pretrained_models.py CodeFormer
xxx替换为本机目录,--input_path为原图片路径,--output_path指定结果图片输出路径
python inference_codeformer.py -w 0.5 --has_aligned --input_path xxx/x.png --output_path xxx
限制上传图片必须为512x512,可通过resize工具裁剪
python inference_colorization.py --input_path xxx/x.png --output_path xxx