一、安装Anaconda
1、Anaconda是什么?
就是可以便捷获取包且对包能够进行管理,同时对环境可以统一管理的发行版本。Anaconda包含了conda、Python在内的超过180个科学包及其依赖项。即它可以在你的电脑上创建多个你想要的python环境,并为每个python环境安装不同的包,不同环境相互切换,操作简单,使用方便!
2、下载安装
官网链接:https://www.anaconda.com/download
安装好后,启动路径在:
***/Anaconda\Scripts\anaconda-navigator-script.py
当然安装好后会自动启动,或者从win里面找也可以。
3、创建python环境
我这里直接用的3.12.7版本
二、.安装PyTorch
1、PyTorch是什么
pytorch是一个开源的Python机器学习库,可以用于支持code former进行机器学习,安装方法如下
2、下载安装
进入Pytorch官网:https://pytorch.org/ 网站,并点击Get Started
在START LOCALLY根据自己电脑配置及安装的软件,选择合适信息即可,这里Computer Platform要看自己电脑显卡适不适合,如果电脑显卡是NVIDIA的,且支持CUDA,则选择对应版本的CUDA即可,如果不是NVIDIA显卡,则选择CPU.
选择完毕之后,将Run this Command的命令行复制一下,到刚刚安装好的Anaconda的codeformer环境中执行一下,等待安装完成即可.注意这里执行从这个界面点进去执行。这样才是当前的环境
注:后续所有命令都是从这个路径进去执行即可。
三、下载源码
这里直接去github下载即可,我这里直接下载压缩包了,懒得git
https://github.com/sczhou/CodeFormer
四、安装依赖
1、安装依赖
直接从Anaconda界面点击刚才创建的环境进到源码解压目录执行
cd G:\AI\CodeFormer\CodeFormer-master
pip3 install -r requirements.txt
2、编译项目
上一步骤执行完后,执行
python basicsr/setup.py develop
3、编译完下载学习模型
分开执行
python scripts/download_pretrained_models.py facelib
python scripts/download_pretrained_models.py dlib
python scripts/download_pretrained_models.py CodeFormer
五、使用测试
还是从步骤四的目录里执行
主要命令
python inserface_codeformer.py -w 0.2 --has_aligned --input_path [image folder]|[image path]
命令说明:
- -w参数:-w是一个权重参数,取值范围为[0,1],取值越小,则图像修复的质量更高,即人脸修复效果最好,取值越大,则会产生保真度更高的图片,就是人脸修复不那么好,但是图片整体效果会好点(实测,建议将w的参数值写成0,这样人脸修复效果最好,参数值写大了,修复结果反倒并不理想)
- –input_path参数:这个就是要修复的图片路径了,可以选单独图片路径,也可以选一个有图片的文件夹,它会将文件夹中,所有的图片都进行修复。
更多命令可以查看github,这里总结了下
1、面部修复
这个命令主要用于面部模糊修复,整个图片脸部占据大部分。
python inference_codeformer.py -w 0.2 --has_aligned --input_path inputs\cropped_faces\0345.png
2、整体图片修复
这个命令用于大图,那种有周围环境的,多人图片
python inference_codeformer.py -w 0.7 --input_path inputs\MyImage\1.png
3、人脸色彩修复
这个主要是,给黑白照片上色
python inference_colorization.py --input_path inputs\gray_faces\Hepburn02.png
4、人脸修复
主要是人脸有遮挡的可以用这个。
python inference_inpainting.py --input_path inputs\masked_faces\00108.png
5、举个例子
可以看到效果还是不错的。再试试上面人脸彩色修复和人脸修复,也是不错的效果。