马上注册,结交更多好友,享用更多功能,让你轻松玩转社区。
您需要 登录 才可以下载或查看,没有账号?立即注册
×
CodeFormer 是一款 AI 视频、图片修复工具,官网项目在 GitHub 的 sczhou/CodeFormer,主要做人脸色彩增强与修复。本文记录在 Mac 上从拉代码、装 brew 和 conda 环境、装依赖、下载预训练模型,到跑人脸修复和图上色两件事的完整流程。
git clone https://github.com/sczhou/CodeFormer
装完在当前目录能看到 CodeFormer 文件夹,进入它:
cd CodeFormer
二、装 conda
Conda 是开源的软件包和环境管理系统,用来装和管理多个版本的包及其依赖。Mac 上可以用 brew 装 anaconda:
brew install anaconda
如果机器上还没有 brew,先装 brew:
/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
装完用下面这条看版本确认:
brew --version
conda 装完还要在 bash 里设 PATH(Mac 芯片的路径是 /opt/homebrew):
export PATH=/opt/homebrew/anaconda3/bin PATH
建一个专属环境:
conda create -n codeformer python=3.8 -y
conda activate codeformer
三、装 Python 依赖
pip3 install -r requirements.txt
python basicsr/setup.py develop
conda install -c conda-forge dlib
四、下载预训练模型
# facelib 和 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
六、几个注意点
- Python 版本锁 3.8,新版 Python 容易装不上老依赖。
- 芯片 Mac 的 conda 路径是 /opt/homebrew/anaconda3,英特尔机型是 /usr/local/anaconda3,别照抄错。
- dlib 走 conda-forge 装比 pip 装省事,避免编译报错。
- 预训练模型体积大,下载慢就挂个梯子或换网络环境。
qidao123.com企服评测-应用市场 编辑提示 |