您当前的位置:首页 > IT编程 > python
| C语言 | Java | VB | VC | python | Android | TensorFlow | C++ | oracle | 学术与代码 | cnn卷积神经网络 | gnn | 图像修复 | Keras | 数据集 | Neo4j | 自然语言处理 | 深度学习 | 医学CAD | 医学影像 | 超参数 | pointnet | pytorch | 异常检测 | Transformers | 情感分类 | 知识图谱 |

自学教程:Python Barbershop实现照片换发型功能

51自学网 2022-02-21 10:36:48
  python
这篇教程Python Barbershop实现照片换发型功能写得很实用,希望能帮到您。

前言

最近看到一个开源项目(Barbershop),可以将照片中的发型更换成另一个,很神奇。先给大家看看项目给出的效果图。

先说说我在安装使用该项目的感受,因为作者给的安装说明太少,我边看代码边安装环境花了整整8个小时,顺便还在等安装的过程中,追了10集电视剧《输赢》。

看在我坚持弄了这么长时间,并把踩过的坑,安装这么长时间的份上,我觉着还是给个赞的。

下面开始我的安装使用记录。

先提供一下Barbershop项目地址:github仓库

环境部署

下面进入最坑的部分,首先我们看一下项目给出的安装说明。

其实只有两条,conda导入environment/environment.yaml环境,下载II2S图片。就这两句话让我装了一晚上,天知道我经历了啥。坑踩了一边,可以按照我的流程装,应该问题不大。

1、导入environment/environment.yaml环境

先创建conda虚拟环境,使用一下命令:

conda env create -f environment/environment.yaml

2、安装pytorch

这里有不会的话,可以看我另一篇文章,有详细的搭建方法:机器学习基础环境部署 

3、依赖库安装

我就不提我踩了多少坑了,下面我把需要的项目没提供的依赖,总结发在下面。

dlib

Ninja

gdown

scikit-image

IPython

opencv-python

版本可以选择最新的,因为我用最新的跑起来了。

4、 cl.exe环境变量配置

需要安装一下Microsoft Visual Studio,我选择的版本是2019版本。

提供一下下载地址:下载 Visual Studio Tools 

这里需要注意安装的时候需要把C++选上。

按照自己的机器位数,选择路径添加到环境变量路径里。下面是我的路径。

添加环境变量

5、模型下载

项目在运行的时候其实是会自己下载模型的,但是从google云盘经常失败,而且该项目要的模型特别多,我放在百度云盘上分享出来,大家可以下我的。

模型文件链接  提取码:tuan

首先人脸68特征数据放到项目的cache目录内。其次另外的6个pt文件放到项目的pretrained_models目录内。

6、发型数据下载

按照项目给出的地址从云盘下载到input/face目录内。这边我还是提供百度云盘的分享链接,下不下来可以用我的。

链接  提取码:TUAN

7、代码调整

项目align_face.py文件中,不会检查模型是否存在,还是会下载,需要调整一下。如下图:

项目验证

OK,环境如果你按照我上面说的部署成功了,恭喜。下面我来验证一下效果。

先看看我准备的照片。

把照片放到unprocessed目录下,注意编号需要和input/face内的文件编号不同。

1、预处理照片

首先我们执行命令预处理unprocessed目录内的照片。命令如下:

python align_face.py
(Barbershop) C:/Users/yi/PycharmProjects/Barbershop>python align_face.pyDownloading Shape Predictor126.jpg: Number of faces detected: 1

可以看到input/face目录多了一个文件,预处理把原照片的人脸处理出来了。

2、换发型

下面看一下我准备换的发型是哪张。

换个短发。执行命令如下:

python main.py --im_path1 126.png --im_path2 94.png --im_path3 94.png --sign realistic --smooth 5

看下执行结果

(Barbershop) C:/Users/yi/PycharmProjects/Barbershop>python main.py --im_path1 126.png --im_path2 94.png --im_path3 94.png --sign realistic --smooth 5Loading StyleGAN2 from checkpoint: pretrained_models/ffhq.ptSetting up Perceptual loss...Loading model from: C:/Users/yi/PycharmProjects/Barbershop/losses/lpips/weights/v0.1/vgg.pth...[net-lin [vgg]] initialized...DoneNumber of images: 2Images: 100%|
python
基于OpenCV的摄像头测距的实现示例
万事OK自学网:51自学网_软件自学网_CAD自学网自学excel、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。