深度学习为图片人物换装【python代码教程】 1
在观看本文之前,请答应我要善良。昨天预告了下,发现很多同学对这个模型都表示出兴趣,甚至有好多同学后台发来照片让我帮他们脱裤子。授人以鱼不如授人以渔,请这些同学好自为之~
01效果演示
本文案例使用的是开源项目instagan,是一种比较新的gan模型建模原理,来自2019年ICLR的论文,
下面看下效果对照:
(出于人道主义,会把人物的长裤脱掉然后换上短裙)
① 原图
② 转换后的图片
02环境配置
首先玩这个模型需要两个前提条件:
有梯子
python3.6版本
有GPU环境(因为源代码是要求必须在GPU的Cuda环境下运行,如果没有GPU的同学推荐用PAI里面的DSWhttps://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):
完成以上步骤,整个环境就搭建好了。
03使用模型做图片转换
注:这个项目的代码有很多hard code的逻辑,需要严格按照下面的做法执行才有可能跑通,包括所有文件的命名。
(1)先设置需要转换的图片
在datasets目录下新增一个test文件夹,构建如下的文件格式
testA和testB存放需要转换的原图,类似于“效果演示中”穿着长裤的图片,testA_seg和testB_seg需要存放mask图片。mask文件是转换图片中的裤子样式,如下图对应实例图片左边的女生裤子(如果想转换其它部位,就不用我举例子了吧):
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能力才能把工程跑起来,小白不太建议使用哈。
深度学习为图片人物换装【python代码教程】 1相关推荐
- 深度学习为图片人物换装【python代码教程】
在观看本文之前,请答应我要善良.昨天预告了下,发现很多同学对这个模型都表示出兴趣,甚至有好多同学后台发来照片让我帮他们脱裤子.授人以鱼不如授人以渔,请这些同学好自为之~ 01效果演示 本文案例使用的是 ...
- python三国演义人物出场统计ppt_Python学习之四大名著人物出场次数Python代码-Go语言中文社区...
<三国演义>,<水浒传>,<西游记>的人物出场次数Python代码: 经过代码运行的结果可以看出三国作者对曹操和孔明比较喜爱:水浒作者对宋江和武松比较喜爱:西游作者 ...
- Python学习之四大名著人物出场次数Python代码
<三国演义>,<水浒传>,<西游记>的人物出场次数Python代码: 经过代码运行的结果可以看出三国作者对曹操和孔明比较喜爱:水浒作者对宋江和武松比较喜爱:西游作者 ...
- python诞生的时间地点人物事件_Python学习之四大名著人物出场次数Python代码
<三国演义>,<水浒传>,<西游记>的人物出场次数Python代码: 经过代码运行的结果可以看出三国作者对曹操和孔明比较喜爱:水浒作者对宋江和武松比较喜爱:西游作者 ...
- 深度学习中的正则化技术(附Python代码)
作者:SHUBHAM JAIN 翻译:和中华 校对:丁楠雅 本文约3500字,建议阅读20分钟. 本文简单介绍了什么是正则化以及在深度学习任务中可以采用哪些正则化技术,并以keras代码具体讲解了一个 ...
- python识别人脸多种属性_深度学习人脸识别仅9行python代码实现?同时高效处理100张相片?...
随着人脸识别.视频结构化等计算视觉相关技术在安防.自动驾驶.手机等领域走向商业化应用阶段,计算视觉技术行业市场迎来大规模的爆发.伴随人脸识别.物体识别等分类.分割算法不断提升精度.计算视觉的核心算法深 ...
- Unity学习笔记:人物换装换武器
知识点:对数据的本地化存储和取用(PlayerPrefs 类) 诞生角色 using UnityEngine; using System.Collections;/// <summary> ...
- 【深度学习之路记录02】python代码批量修改Labelme标注的json文件(删除标签、修改标签名)
代码参考:https://blog.csdn.net/qq_44442727/article/details/112785978 创建自己的数据集时,经常需要一些调整,比如说修改某一批文件中已经标好的 ...
- 深度学习狗图片_狗品种分类的深度学习
深度学习狗图片 深度学习 (Deep Learning) Stuck behind the paywall? Click here to read the full story with my fri ...
最新文章
- 成功解决sys.argv[1] IndexError: list index out of range错误
- 十一、深入Java的判断语句
- 简单实现仿某宝地址选择三级联动样式
- 微信多开txt_电脑版微信怎么双开、多开
- 洛谷 P2919 [USACO08NOV]守护农场Guarding the Farm
- 如何查询linux服务器的网卡,Linux服务器如何查看有没有无线网卡
- Windows下如何生成数字证书
- ubuntu 16.04 R语言安装 rJava package失败解决方案
- java 短链接url_Java 网址短链接服务原理及解决方案
- 玩转树莓派-2.配置你的树莓派
- java框架常见的面试题
- 经典Hive-SQL面试题及答案
- CPL、BCPL、B语言
- 看中国魅力女强人 访格力电器总裁董明珠
- 阿里巴巴官方:明年今日马云将不再担任董事局主席
- Kylin Error:Cannot start job scheduler due to lack of job lock
- [乡土民间故事_徐苟三传奇]第四四回_赵员外苕吃假香肠
- android内存扩展格式,一文看懂:如何为你的安卓手机选择合适的存储卡
- 循环里发送请求java_servlets – 在Java Servlet中转发请求时的无限循环
- 基址寻址和变址寻址区别(白话版)