老照片修复工具CodeFormer WebUI 轻松本地一键部署(已提供最新完整工具包)
CodeFormer是一种使用人工智能修复旧照片或低质量人脸图像的开源程序,针对于老照片处理就是把低质量,小尺寸,有残缺的老的照片进行修复,主要途径有上色、破损修复、放大完整修复,整体效果非常的棒。
但是由于开源的CodeFormer时间上有点旧,代码有点旧,部署的环境有点复杂,所以本地要成功运行变得比较的困难,今天就经过修正代码后的CodeFormer本地独立安装部署的方法记录一下,分享给大家!
CodeFormer概述
Github主页地址:https://github.com/sczhou/CodeFormer
开发机构:南洋理工大学S实验室
项目参与者:
系统环境准备
1. 安装 python3.10.6
确保你的python是3.10.6,如果不是去官网下载:
https://www.python.org/ftp/python/3.10.6/python-3.10.6-amd64.exe
2.安装Cmake工具
3.安装好系统运行库
https://github.com/abbodi1406/vcredist/releases
4. 安装pip
pip 是python的包管理器,用这个去安装各种依赖包
cmd或Powershell里面运行安装命令
python -m ensurepip
在软件开发中,依赖包通常是指一种软件库或模块所依赖的其他软件库或模块,也可以称之为依赖项。
举个栗子,假设您正在开发一个名为“炒青菜”的程序,该程序的输入是青菜、油、盐,输出是炒好的青菜。在这个例子中,您的程序依赖于以下几个依赖包:
- 一个能够处理青菜的库:该库可以提供清洗、切菜等功能。
- 一个能够处理油的库:该库可以提供加热、放油等功能。
- 一个能够处理盐的库:该库可以提供加盐等功能。
- 一个能够将食材烹饪成菜肴的库:该库可以提供炒菜等功能。
如果没有这些依赖包,您的“炒青菜”程序将无法正常运行。
依赖包在软件开发中非常重要,因为它们可以帮助开发人员避免重复造轮子,提高开发效率。此外,依赖包还可以确保软件的质量和稳定性,因为它们通常经过了充分的测试和验证。
5.安装Pytorch和对应的CUDA
NVIDIA 驱动程序下载
https://www.nvidia.cn/Download/index.aspx?lang=cn
Pytorch和对应的CUDA安装
https://pytorch.org/get-started/previous-versions/
详细查看本机的GPU信息,请访问之前的文章
https://mn.cyou/archives/1709116271511
nvidia-smi
nvcc --version
安装依赖
cmd或Powershell里面运行安装命令
python.exe -m pip install --upgrade pip
pip install -r ./requirements.txt
如果你的网络不好,可以用下面这条加速安装
pip install -r ./requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
安装模型和其他资源
git clone https://github.com/davisking/dlib.git
cd dlib
python setup.py install
pip install gradio
下载模型
python scripts/download_pretrained_models.py facelib
python scripts/download_pretrained_models.py dlib
python scripts/download_pretrained_models.py CodeFormer
创建虚拟环境
防止以后python工程多了,依赖关系混乱,如果你的python只做这一个事情,可以忽略
虚拟环境就是基于当前文件夹路径进行python依赖构建,这样对以后的其他工程没有影响,当然其他工程对他也没有影响
-
创建一个叫venv虚拟环境
cmd或Powershell里面运行安装命令
python -m venv venv
-
激活并使用他 cmd命令行运行:
call venv/Scripts/activate.bat
-
PowerShell环境运行:
venv/Scripts/activate
运行UI
直接运行文件夹下CodeFormer运行webui.bat文件
效果
CUDA
可以直接检测本地CUDA环境
照片划痕修复
先用别的软件把破损的要修复的地方涂白,打开软件导入处理就可以了
黑白照片上色
如果照片本身是黑白的可以通过这个上色处理
整体放大修复
直接默认参数就行,可以放大修复到成品,效果突出
在线老照片修复
CodeFormer
https://huggingface.co/spaces/sczhou/CodeFormer
GFPGAN
https://huggingface.co/spaces/akhaliq/GFPGAN
在线生成照片效果对比
一键部署包下载地址
文件路径:科技美南/AI/CodeFormer.7z
使用一键部署包只需要注意运行以下几个步骤:
安装好系统的运行库
安装好python3.10.6
有N卡的安装Pytorch和对应的CUDA
python.exe -m pip install --upgrade pip
pip install -r ./requirements.txt
如果你的网络不好,可以用下面这条加速安装
pip install -r ./requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
pip install gradio
评论区