mac环境下安装Codeformer教程

github地址：https://github.com/sczhou/CodeFormer#face-color-enhancement-and-restoration

CodeFormer 是一款AI 视频、图片修复软件。

1.下载代码

git clone https://github.com/sczhou/CodeFormer

安装完成后可在当前目录看到：CodeFormer

2.进入CodeFormer

cd CodeFormer

3.安装conda

Conda 是一个开源的软件包管理系统和环境管理系统，用于安装和管理多个不同版本的软件包和它们的依赖关系

brew install anaconda

3.1安装brew

如果没有brew，还需要安装brew

/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"

安装完成可以通过如下命令查看：

brew --version

3.2 conda PATH

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

4.安装python依赖

执行如下三个命令

pip3 install -r requirements.txt
python basicsr/setup.py develop
conda install -c conda-forge dlib

5.下载训练模型

# 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

6.工具使用

6.1图形修复

xxx替换为本机目录，--input_path为原图片路径，--output_path指定结果图片输出路径

python inference_codeformer.py -w 0.5 --has_aligned --input_path xxx/x.png --output_path xxx

6.2图形上色

限制上传图片必须为512x512，可通过resize工具裁剪

python inference_colorization.py --input_path xxx/x.png --output_path xxx