Python Barbershop实现照片换发型功能

更新时间：2022年01月04日 09:02:21 作者：剑客阿良_ALiang

这篇文章主要为大家介绍了一个开源项目（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.py
Downloading Shape Predictor
126.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 5
Loading StyleGAN2 from checkpoint: pretrained_models/ffhq.pt
Setting up Perceptual loss...
Loading model from: C:\Users\yi\PycharmProjects\Barbershop\losses\lpips\weights\v0.1\vgg.pth
...[net-lin [vgg]] initialized
...Done
Number of images: 2
Images: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 2/2 [10:39<00:00, 319.74s/it]
Number of images: 2
Images: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 2/2 [02:08<00:00, 64.04s/it]
Loading StyleGAN2 from checkpoint: pretrained_models/ffhq.pt
C:\Users\yi\PycharmProjects\Barbershop\models\Alignment.py:97: UserWarning: where received a uint8 condition tensor. This behavior is deprecated and will be removed in a
 future version of PyTorch. Use a boolean condition instead. (Triggered internally at  ..\aten\src\ATen\native\TensorCompare.cpp:328.)
  new_target_final = torch.where(OB_region, torch.from_numpy(new_target_inpainted), new_target)
Align Step 2:   0%|                                                                                                                             | 0/100 [00:00<?, ?it/s]E
:\ProgramData\Anaconda3\envs\Barbershop\lib\site-packages\torch\nn\functional.py:3680: UserWarning: The default behavior for interpolate/upsample with float scale_factor
 changed in 1.6.0 to align with other frameworks/libraries, and now uses scale_factor directly, instead of relying on the computed output size. If you wish to restore th
e old behavior, please set recompute_scale_factor=True. See the documentation of nn.Upsample for details.
  "The default behavior for interpolate/upsample with float scale_factor changed "
E:\ProgramData\Anaconda3\envs\Barbershop\lib\site-packages\torch\nn\functional.py:3635: UserWarning: Default upsampling behavior when mode=bicubic is changed to align_co
rners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details.
  "See the documentation of nn.Upsample for details.".format(mode)
Loading StyleGAN2 from checkpoint: pretrained_models/ffhq.pt
Setting up Perceptual loss...
Loading model from: C:\Users\yi\PycharmProjects\Barbershop\losses\masked_lpips\weights\v0.1\vgg.pth
...[net-lin [vgg]] initialized
...Done
Setting up Perceptual loss...
Loading model from: C:\Users\yi\PycharmProjects\Barbershop\losses\masked_lpips\weights\v0.1\vgg.pth
...[net-lin [vgg]] initialized
...Done

等了很长时间，结果出来了。最后的效果图如下

对比一下看看，还是很不错的吧，我这么长的安装时间没有白费。

总结

该项目最终的效果是非常好的，只是安装的过程比较繁琐。而且在我跑最终融合的效果时，我似乎都快闻到显卡的焦味了。花费了很多时间，解决了很多问题，也有了很多收获，成长不外如此。

以上就是Python Barbershop实现照片换发型功能的详细内容，更多关于Python Barbershop照片换发型的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

关于Python中浮点数精度处理的技巧总结
双精度浮点数(double)是计算机使用的一种数据类型，使用 64 位（8字节）来存储一个浮点数。下面这篇文章主要给大家总结介绍了关于Python中浮点数精度处理的技巧，需要的朋友可以参考借鉴，下面来一起看看吧。
2017-08-08
wxpython绘制音频效果
这篇文章主要为大家详细介绍了wxpython绘制音频效果，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2019-11-11
Python特征降维知识点总结
在本篇文章里小编给大家整理了一篇关于Python特征降维知识点总结内容，有需要的朋友们可以学习参考下。
2021-08-08
GitHub AI编程工具copilot在Pycharm的应用
最近听说github出了一种最新的插件叫做copilot,这篇文章主要给大家介绍了关于GitHub AI编程工具copilot在Pycharm的应用,目前感觉确实不错,建议大家也去使用,需要的朋友可以参考下
2022-04-04
python连接oracle数据库操作实例及遇到的异常
这篇文章主要给大家介绍了关于python连接oracle数据库操作实例及遇到的一些异常,Oracle是著名且广泛使用的数据库之一,而python的数据处理功能可以很好地利用这种连接性,文中通过代码介绍的非常详细,需要的朋友可以参考下
2023-12-12
python3.8+django2+celery5.2.7环境准备(python测试开发django)
这篇文章主要介绍了python测试开发django之python3.8+django2+celery5.2.7环境准备工作，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
2022-07-07
让python在hadoop上跑起来
让python在hadoop上跑起来，python如何在hadoop上跑起来？感兴趣的小伙伴们可以参考一下
2016-01-01
python实现图片压缩代码实例
这篇文章主要介绍了python实现图片压缩代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-08-08
Python对列表排序的方法实例分析
这篇文章主要介绍了Python对列表排序的方法,实例分析了Python列表排序函数的相关使用技巧,非常简单实用,需要的朋友可以参考下
2015-05-05
Python对字符串实现去重操作的方法示例
字符串去重是python中字符串操作常见的一个需求，最近在工作中就又遇到了，所以下面这篇文章主要给大家介绍了关于Python对字符串实现去重操作的相关资料，文中给出了详细的介绍，需要的朋友可以参考借鉴，下面来一起看看吧。
2017-08-08