深度学习为图片人物换装【python代码教程】

栏目: Python · 发布时间: 5年前

内容简介:在观看本文之前,请答应我要善良。昨天预告了下,发现很多同学对这个模型都表示出兴趣,甚至有好多同学后台发来照片让我帮他们脱裤子。授人以鱼不如授人以渔,请这些同学好自为之~本文案例使用的是开源项目instagan,是一种比较新的gan模型建模原理,来自2019年ICLR的论文,下面看下效果对照:

在观看本文之前,请答应我要善良。昨天预告了下,发现很多同学对这个模型都表示出兴趣,甚至有好多同学后台发来照片让我帮他们脱裤子。授人以鱼不如授人以渔,请这些同学好自为之~

01.效果演示

本文案例使用的是开源项目instagan,是一种比较新的gan模型建模原理,来自2019年ICLR的论文,

下面看下效果对照:

(出于人道主义,会把人物的长裤脱掉然后换上短裙)

深度学习为图片人物换装【python代码教程】

深度学习为图片人物换装【python代码教程】

02.环境配置

首先玩这个模型需要两个前提条件:

  1. 有梯子
  2. python3.6版本
  3. 有GPU环境(因为源代码是要求必须在GPU的Cuda环境下运行,如果没有GPU的同学推荐用PAI里面的DSW https://data.aliyun.com/product/learn  ,比较便宜 )

(1)下载实验代码: https://github.com/sangwoomo/instagan

(2)下载pre-trained model,如果只是实验就直接用训练好的模型即可: https://drive.google.com/drive/folders/1xb9rR21MhMVselc6HTmOr73WOkOviFmO

(如果只玩换裤子这个实验,下载pants2skirt_mhp_instagan这个模型即可)

(3)安装代码中的requirement.txt中的依赖包

最后把下载的model文件放到代码文件的根目录下,目录结构如下(蓝色部分为模型文件夹,里面是200_net_G_A.pth和200_net_G_B.pth):

深度学习为图片人物换装【python代码教程】

完成以上步骤,整个环境就搭建好了。

03.使用模型做图片转换

注:这个项目的代码有很多hard code的逻辑,需要严格按照下面的做法执行才有可能跑通,包括所有文件的命名。

(1)先设置需要转换的图片

在datasets目录下新增一个test文件夹,构建如下的文件格式

深度学习为图片人物换装【python代码教程】

testA和testB存放需要转换的原图,类似于“效果演示中”穿着长裤的图片,testA_seg和testB_seg需要存放mask图片。mask文件是转换图片中的裤子样式,如下图对应实例图片左边的女生裤子(如果想转换其它部位,就不用我举例子了吧):

深度学习为图片人物换装【python代码教程】

mask图片还需要与被转换图片命名一致,详细规则参见源代码中的datasets,如果不一致会出现以下错误: https://github.com/sangwoomo/instagan/issues/5

(2)执行图片转换逻辑

在工程的根目录下执行以下代码,使用工程下的test.py这个测试代码:

 python test.py --dataroot ./datasets/test --model instagan --name pants2skirt_mhp_instagan --loadSizeH 240 --loadSizeW 160 --fineSizeH 240 --fineSizeW 160 --ins_per 2 --ins_max 20

如果没有报错的话在GPU环境下1分钟就可以执行完毕,代码执行完毕后在工程的results文件夹下就能看到转换好的图片了。

04.总结

整个项目的效果还是不错的,不过可能代码还没来得及优化,可能在实验的过程中有很多坑,大家需要一定的debug能力才能把工程跑起来,小白不太建议使用哈。

本文作者:傲海

阅读原文

本文为云栖社区原创内容,未经允许不得转载。


以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

Google御用網頁語言Node.js

Google御用網頁語言Node.js

郭家寶 / 佳魁資訊 / 2013-4-26 / NT 490

這是一本 Node.js 的入門教學,寫給想要學習 Node.js,但沒有任何系統的經驗的開發者。如果你聽說過 Node.js,並被它許多神奇的特性吸引,本書就是為你準備的。 透過閱讀本書,你可以對 Node.js 有全面的認識,學會如何用 Node.js 程式設計,了解事件驅動、非同步式 I/O 的程式設計模式,同時還可以了解一些使用JavaScript 進行函數式程式設計的方法。 ......一起来看看 《Google御用網頁語言Node.js》 这本书的介绍吧!

HTML 压缩/解压工具
HTML 压缩/解压工具

在线压缩/解压 HTML 代码

URL 编码/解码
URL 编码/解码

URL 编码/解码

HSV CMYK 转换工具
HSV CMYK 转换工具

HSV CMYK互换工具