用AI帮你画新年贺卡:只需输入几个单词,就能模仿大师名作

晓查 发自 凹非寺

量子位 出品 | 公众号 QbitAI

如果你的手法拙劣,没有任何艺术细菌,自己作画完全无法见人。但是你想给妹子送上一张有个性的新年贺卡怎么办?

虽然没有梵高、毕加索的才华,但你能让AI学会做这件事啊!

用AI帮你画新年贺卡:只需输入几个单词,就能模仿大师名作

有了这个ArtistAI,你甚至连简笔画都不用会,只要轻松地在键盘上敲击几下,就能制作一张名家风格的个性化电子贺卡。

让大师为你画画

无论是什么物体、什么风格,ArtistAI统统能拿下。

告诉AI画一个篮子,再给它一张表现主义大师蒙克的名画《尖叫》,就能得到这样一张贺卡:

用AI帮你画新年贺卡:只需输入几个单词,就能模仿大师名作

又或者是你想让大师毕加索为你画一朵花:

用AI帮你画新年贺卡:只需输入几个单词,就能模仿大师名作

都能轻松搞定。

从单词到贺卡

ArtistAI内集成了NLP、图像生成、风格迁移等技术,3步内解决问题。

首先,我们必须要让ArtistAI的脑洞足够大,无论输入什么抽象单词,总能想到要画什么。

为此,需要使用GloVe单词嵌入模型,用维基百科词条进行训练。将输入的单词和谷歌“猜画小歌”里的图片分类进行比较,找出最贴近的分类。

不管你输入什么,ArtistAI总能联想到一个“猜画小歌”里合适的分类。

用AI帮你画新年贺卡:只需输入几个单词,就能模仿大师名作

但有时它也不太聪明,比如输入美国演员“Chris Pine”,居然会联想到“树”。所以你还是对它直接一点,别太刁难了。

第二步,让ArtistAI向“猜画小歌”学会涂鸦。

在上一步中,我们让它脑洞大开想到了“树”,接着使用名为Sketch-RNN的变分自动编码器(VAE)从数据集学习足够多的涂鸦,然后ArtistAI就能画出自己的简笔画了。

用AI帮你画新年贺卡:只需输入几个单词,就能模仿大师名作

最后一步,我们训练ArtistAI将涂鸦变成有名家风格的画作。

这一步中,使用了神经风格迁移技术。AI尝试理解特定画家的笔触和调色风格,并将这种风格迁移到刚才的涂鸦上。

用AI帮你画新年贺卡:只需输入几个单词,就能模仿大师名作

使用教程

ArtistAI目前已经在Github上开源了,虽然叫christmAIs,为圣诞贺卡而生,但是马上新年也能用上啊。

它的安装过程非常简单。直接从GitHub上拉取项目,甚至可以傻瓜一键式安装:

source venv/bin/activate # Highly recommended

./install-christmais.sh

通过两行命令直接安装好了依赖的软件库和风格迁移模型。

然后在终端里用Python运行,命令中各项参数如下:

python -m christmais.tasks.christmais_time

–input=<输入某个单词>

–style=<你喜欢的名画所在路径>

–output=<输出图片文件路径>

这样就能获得了一张艺术风格贺卡,是不是很容易?

传送门

原文:

https://stories.thinkingmachin.es/ai-art-holiday-cards/

项目地址:

https://github.com/thinkingmachines/christmAIs

版权所有,未经授权不得以任何形式转载及使用,违者必究。