您现在的位置是:主页 > Web前端技术 > Web前端技术

Python中Barbershop如何实现照片换发型功能开发技术

IDCBT2022-01-05服务器技术人已围观

简介这篇文章主要介绍Python中Barbershop如何实现照片换发型功能,文中介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们一定要看完! 环境部署 下面进入最坑的部分,首先我们看一

这篇文章主要介绍Python中Barbershop如何实现照片换发型功能,文中介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们一定要看完!

      环境部署

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

      其实只有两条,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

      标签:

      很赞哦! ()

本栏推荐