目 录CONTENT

文章目录

老照片修复工具CodeFormer WebUI 轻松本地一键部署(已提供最新完整工具包)

科技美南
2024-04-24 / 0 评论 / 1 点赞 / 221 阅读 / 6109 字 / 正在检测是否收录...
温馨提示:
本文最后更新于 2024-05-02,若内容或图片失效,请留言反馈。部分素材来自网络,若不小心影响到您的利益,请联系我们删除。

老照片修复工具CodeFormer WebUI 轻松本地一键部署(已提供最新完整工具包)

CodeFormer是一种使用人工智能修复旧照片或低质量人脸图像的开源程序,针对于老照片处理就是把低质量,小尺寸,有残缺的老的照片进行修复,主要途径有上色、破损修复、放大完整修复,整体效果非常的棒。

但是由于开源的CodeFormer时间上有点旧,代码有点旧,部署的环境有点复杂,所以本地要成功运行变得比较的困难,今天就经过修正代码后的CodeFormer本地独立安装部署的方法记录一下,分享给大家!

CodeFormer概述

Github主页地址:https://github.com/sczhou/CodeFormer

开发机构:南洋理工大学S实验室

项目参与者:

20231007091450s.jpg

20231007091425s.jpg

20231007091525s.jpg

20231007091553s.jpg

系统环境准备

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工具

https://cmake.org/download/

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文件

微信截图_20240424095314.jpg

微信截图_20240424095343.jpg

效果

CUDA

可以直接检测本地CUDA环境

微信截图_20240424101441.jpg

照片划痕修复

先用别的软件把破损的要修复的地方涂白,打开软件导入处理就可以了

微信截图_20240424101507.jpg

黑白照片上色

如果照片本身是黑白的可以通过这个上色处理

微信截图_20240424101531.jpg

整体放大修复

直接默认参数就行,可以放大修复到成品,效果突出

微信截图_20240424101639.jpg

在线老照片修复

CodeFormer

https://huggingface.co/spaces/sczhou/CodeFormer

20240424141539.jpg

GFPGAN

https://huggingface.co/spaces/akhaliq/GFPGAN

微信截图_20240424144239.jpg

在线生成照片效果对比

https://imgsli.com/MjU4NTU2

微信截图_20240424145000.jpg

一键部署包下载地址

文件路径:科技美南/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

视频教程在线观看

1

评论区