您现在的位置是:主页 > 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.yaml2、安装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标签:
很赞哦! ()