今天小李在逛人类知识宝库时发现一个非常有意思的项目,“CoupleGenerator”, 可以理解为伴侣生成器。哦也不对,单身的小伙伴不要想太多,国家不会分配,AI自然目前也是没法大变活人,给你分配对象的。
那这个项目是做什么的呢,准确的说是生成伴侣的样貌图片,通过一定夫妻合照的数据集进行训练,你只要输入你的照片便可以通过生成对抗网络训练出一个和你最为般配的你的“爱人”的图片。下图是模型训练了8800个步骤之后的结果,
环境配置
- tensorflow 1.1
- vgg 的预训练权重 (为了方便大家,我已经训练好一个,文末可下载)
快速开始
- 下载训练图片并且解压缩:https://cloudstor.aarnet.edu.... (这里要注意的是原作者把训练图片反向命名了,命名成了test)
- 克隆作者的Github仓库
git clone https://github.com/irfanICMLL/CoupleGenerator
- 运行 autotest.sh 脚本
“这是两年前的旧项目,她非常惊讶那么多人会喜欢这个玩具项目”,作者在README上说道,“数据源来自网络爬虫所获取的已婚夫妇的照片,并对想进行预处理,使用了 pix2pix 的代码进行模型的训练。”
这个爬虫是一个百度图片的小爬虫。看得出来,21世纪的今天最重要的以及必须的就是数据,我们需要掌握一定的数据获取的能力。我们知道,深度学习是由数据驱动的,也非常依赖于数据集,如果能收集到更多的训练图片,伴侣生成的效果会变得更加出色。
那么大家还在等什么?没有对象的去GAN一个Couple,说不定人海茫茫你就能遇见那个他;有对象的去GAN一个小三,看好了什么长相最有可能插足你们之间的感情!
哦对了,这里是已经训练好的vgg16模型。
链接:https://pan.baidu.com/s/1NuKd...
提取码:ocnl
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。