CodeFormer WebUI 牛哥版本一键部署 | 免费下载

写在最前:

老照片处理就是把 低质量小尺寸有残缺,的老的照片进行修复。

上色 -> 破损修复 -> 放大完整修复

CodeFormer本地部署这个视频发布以后,来自小伙伴的反馈就多了起来:
CodeFormer WebUI 牛哥版本一键部署 | 免费下载

概述一下:

  • 弄个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,看这里:
    CodeFormer WebUI 牛哥版本一键部署 | 免费下载

    • CMD 模式下:
      CodeFormer WebUI 牛哥版本一键部署 | 免费下载

    • PowerShell下:
      CodeFormer WebUI 牛哥版本一键部署 | 免费下载

    推荐使用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

    CodeFormer WebUI 牛哥版本一键部署 | 免费下载

    CodeFormer WebUI 牛哥版本一键部署 | 免费下载

    UI介绍:

    整体分三部分,分别对应CodeFormer官方的 放大 | 上色 | 破损修复
    – 对接了每个功能可以支持的所有参数
    – 默认大部分参数不用管,牛哥对默认值做了调整,除了输入输出目录,其他如无必要都不需要动
    – 点击处理图片按钮就可以开始处理了
    – 完成后图像展示区会展示结果

    BTW:所有的AI都是依赖于N卡的硬件加速,即:CUDA的,当然,使用CPU等模式也可以做到能运行,但是会很慢,所以,要玩AI:
    – 有一个Nvidia的显卡是很有必要的,推荐:4060ti 16GB(速度OK,有16GB显存,可以小规模炼丹了)
    – 确保相关环境是OK的,比如本文内需要CUDA加速就会快很多,要想让其顺利依赖cuda良好运行,python的依赖库版本要和cuda的版本做好对应

    牛哥在webUI里提供了一键获取CUDA信息的能力:
    CodeFormer WebUI 牛哥版本一键部署 | 免费下载
    注意这里的 cu121字样,忽略数字,这个cu就是代表显卡硬件加速在我们的工程里正常了,这样才能快,如果你看到的是cpu:
    CodeFormer WebUI 牛哥版本一键部署 | 免费下载
    那就做好速度超级慢的打算吧。。。大概是CUDA加速的10倍时间。。。

    更多的关于显卡硬件加速的能力配置,参考牛哥之前的视频:
    Jan-本地部署

    效果:

    CUDA:

    为了让大家体验效果好,这里封装了一个检测能力,上午已经说过了,这里就略过了
    CodeFormer WebUI 牛哥版本一键部署 | 免费下载

    照片划痕修复:

    1. 先用别的软件把破损的要修复的地方涂白(这个以后可以考虑集成处理)
    2. 打开软件导入处理就可以

    CodeFormer WebUI 牛哥版本一键部署 | 免费下载

    黑白照片上色:

    如果照片本身是黑白的是需要这一步的:
    CodeFormer WebUI 牛哥版本一键部署 | 免费下载

    整体放大修复:

    最后就是整体放大修复出图了,这里参数比较多,直接做成文件夹处理了,默认参数牛哥都调整好了

    CodeFormer WebUI 牛哥版本一键部署 | 免费下载

    写在最后

    小结

    • 以上AI相关功能的web-ui封装已经处理完毕了,后续暂时就不多花精力了;
    • 这可能是关于CodeFormer最稳定的封装;
    • CodeFormer本身依然存在不足,比如对环境的修补等
    • 后续会结合神经网络控制以及SD的一些模型做一些工作流封装(预计会放到CloseAI里,没关注的小伙伴赶紧关注哦)

    资源

    本文相关修改的源码放到这个压缩包了:codeformer-niugee.web-ui.patch_.zip

    123网盘
    https://www.123pan.com/s/r11hTd-b2wsv.html
    提取码:NIGG

    Google 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, 按下图把这句代码注释掉,注释的方法:在最前输入#号

    CodeFormer WebUI 牛哥版本一键部署 | 免费下载

站内快搜



联系牛哥

加入牛哥的 TG 群组一起交流

=>>Telegram群组(点我加入)

有特殊情况的,需要单独沟通,可以添加牛哥的 微信 或者 Telegram