写在最前:
老照片处理就是把 低质量,小尺寸,有残缺,的老的照片进行修复。
上色 -> 破损修复 -> 放大完整修复
从 CodeFormer本地部署这个视频发布以后,来自小伙伴的反馈就多了起来:
概述一下:
- 弄个UI:比如gradio,方便分享
- 简化,简化,在再简化,否则搞不定
结论:
– 牛哥对原有代码做了修改和调整,然后构建了一个一键启动,构建webUI
前置环境:
1. 安装 python3.10.6
确保你的python是3.10.6,如果不是去官网下载:
https://www.python.org/ftp/python/3.10.6/python-3.10.6-amd64.exe
2. 安装pip
pip 是python的包管理器,用这个去安装各种依赖包
所谓依赖包,直白的理解就是,运行程序的时候需要的其他资源,比如你写了个word文档,要使用他需要你机器有word程序;如果文档就是这个程序,那么word就是他的依赖包(不知道这么说能不能理解。。。)
shell
python install pip
3. 创建虚拟环境:
防止以后python工程多了,依赖关系混乱,如果你的python只做这一个事情,可以忽略
虚拟环境就是基于当前文件夹路径进行python依赖构建,这样对以后的其他工程没有影响,当然其他工程对他也没有影响
- 创建一个叫venv虚拟环境
cmd
python -m venv venv
- 激活并使用他
cmd 如果是 cmd 命令执行: call venv/Scripts/activate.bat 如果是PowerShell环境,执行: venv/Scripts/activate
windows 11 提供了很多运行命令行的能力
– CMD :最初的命令行,从win98那个年代就有了,老,丑。。。
– PowerShell :忘记是win7 还是 哪个版本开始带的了,为了兼容类似linux的shell能力
– 终端 :win11特有的,漂亮,把上面都集成了,可以运行 cmd模式,也可以powershell模式win11 下如果安装了终端,默认都是由他作为交互界面,如何确定是CMD还是PowerShell,看这里:
CMD 模式下:
PowerShell下:
推荐使用PowerShell模式的终端作为交互界面,无他,功能强大,好看
安装依赖:
就一句话:
shell pip install -r ./requirements.txt
这个文件里面把依赖包都加进去了,直接安装就好
如果,你的网络不好(处在一些伟大的防火墙内),那么你可以换个快的源,执行这条命令:
shell pip install -r ./requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
安装模型等其他资源
shell 提前安装好 cmake pip install dlib 下载模型 python scripts/download_pretrained_models.py facelib python scripts/download_pretrained_models.py dlib python scripts/download_pretrained_models.py CodeFormer
- 下载模型有点慢,别着急。。。。
- 安装dlib需要本地有cmake环境(动态编译需要),所以需要去安装一个Virtual Studio 20XX
这一块不展开了,有不会的去看我之前介绍环境安装的详细视频:
CodeFormer 本地部署(上)
运行UI:
经过以上,直接运行这个:
shell ./niugee.ui.bat 或者: python codeformer.niugee-ui.py
UI介绍:
整体分三部分,分别对应CodeFormer官方的 放大 | 上色 | 破损修复
– 对接了每个功能可以支持的所有参数
– 默认大部分参数不用管,牛哥对默认值做了调整,除了输入输出目录,其他如无必要都不需要动
– 点击处理图片按钮就可以开始处理了
– 完成后图像展示区会展示结果BTW:所有的AI都是依赖于N卡的硬件加速,即:CUDA的,当然,使用CPU等模式也可以做到能运行,但是会很慢,所以,要玩AI:
– 有一个Nvidia的显卡是很有必要的,推荐:4060ti 16GB(速度OK,有16GB显存,可以小规模炼丹了)
– 确保相关环境是OK的,比如本文内需要CUDA加速就会快很多,要想让其顺利依赖cuda良好运行,python的依赖库版本要和cuda的版本做好对应牛哥在webUI里提供了一键获取CUDA信息的能力:
注意这里的 cu121字样,忽略数字,这个cu就是代表显卡硬件加速在我们的工程里正常了,这样才能快,如果你看到的是cpu:
那就做好速度超级慢的打算吧。。。大概是CUDA加速的10倍时间。。。更多的关于显卡硬件加速的能力配置,参考牛哥之前的视频:
Jan-本地部署效果:
CUDA:
为了让大家体验效果好,这里封装了一个检测能力,上午已经说过了,这里就略过了
照片划痕修复:
- 先用别的软件把破损的要修复的地方涂白(这个以后可以考虑集成处理)
- 打开软件导入处理就可以
黑白照片上色:
如果照片本身是黑白的是需要这一步的:
整体放大修复:
最后就是整体放大修复出图了,这里参数比较多,直接做成文件夹处理了,默认参数牛哥都调整好了
写在最后
小结
- 以上AI相关功能的web-ui封装已经处理完毕了,后续暂时就不多花精力了;
- 这可能是关于CodeFormer最稳定的封装;
- CodeFormer本身依然存在不足,比如对环境的修补等
- 后续会结合神经网络控制以及SD的一些模型做一些工作流封装(预计会放到CloseAI里,没关注的小伙伴赶紧关注哦)
资源
本文相关修改的源码放到这个压缩包了:codeformer-niugee.web-ui.patch_.zip
123网盘
https://www.123pan.com/s/r11hTd-b2wsv.html
提取码:NIGGGoogle Drive
https://drive.google.com/drive/folders/1-AbrPv09a7YRTYM4Od_Byvc8UqqQj0J1?usp=drive_link百度云
链接:https://pan.baidu.com/s/1pWaziAC7xMV908TuOkYdyw?pwd=niug
提取码:niug使用:
1. 下载官方源码并做好初始工作(参见之前视频)
2. 解压这个压缩包,把所有文件拷贝到官方源码根目录,如果遇到替换提示,点是!遇到提示 basicsr-version-error, 按下图把这句代码注释掉,注释的方法:在最前输入#号