个人随笔
目录
CodeFormar:面部修复,人脸修复,人脸上色教程
2024-10-14 20:55:01

一、安装Anaconda

1、Anaconda是什么?

就是可以便捷获取包且对包能够进行管理,同时对环境可以统一管理的发行版本。Anaconda包含了conda、Python在内的超过180个科学包及其依赖项。即它可以在你的电脑上创建多个你想要的python环境,并为每个python环境安装不同的包,不同环境相互切换,操作简单,使用方便!

2、下载安装

官网链接:https://www.anaconda.com/download
安装好后,启动路径在:

  1. ***/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界面点击刚才创建的环境进到源码解压目录执行

  1. cd G:\AI\CodeFormer\CodeFormer-master
  2. pip3 install -r requirements.txt

2、编译项目

上一步骤执行完后,执行

  1. python basicsr/setup.py develop

3、编译完下载学习模型

分开执行

  1. python scripts/download_pretrained_models.py facelib
  2. python scripts/download_pretrained_models.py dlib
  3. python scripts/download_pretrained_models.py CodeFormer

五、使用测试

还是从步骤四的目录里执行
主要命令

  1. 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、面部修复

这个命令主要用于面部模糊修复,整个图片脸部占据大部分。

  1. python inference_codeformer.py -w 0.2 --has_aligned --input_path inputs\cropped_faces\0345.png

2、整体图片修复

这个命令用于大图,那种有周围环境的,多人图片

  1. python inference_codeformer.py -w 0.7 --input_path inputs\MyImage\1.png

3、人脸色彩修复

这个主要是,给黑白照片上色

  1. python inference_colorization.py --input_path inputs\gray_faces\Hepburn02.png

4、人脸修复

主要是人脸有遮挡的可以用这个。

  1. python inference_inpainting.py --input_path inputs\masked_faces\00108.png

5、举个例子

可以看到效果还是不错的。再试试上面人脸彩色修复和人脸修复,也是不错的效果。





 15

啊!这个可能是世界上最丑的留言输入框功能~


当然,也是最丑的留言列表

有疑问发邮件到 : suibibk@qq.com 侵权立删
Copyright : 个人随笔   备案号 : 粤ICP备18099399号-2