如果你的手法拙劣,没有任何艺术细菌,自己作画完全无法见人。但是你想给妹子送上一张有个性的新年贺卡怎么办?
虽然没有梵高、毕加索的才华,但你能让AI学会做这件事啊!
有了这个ArtistAI,你甚至连简笔画都不用会,只要轻松地在键盘上敲击几下,就能制作一张名家风格的个性化电子贺卡。
让大师为你画画
无论是什么物体、什么风格,ArtistAI统统能拿下。
告诉AI画一个篮子,再给它一张表现主义大师蒙克的名画《尖叫》,就能得到这样一张贺卡:
又或者是你想让大师毕加索为你画一朵花:
都能轻松搞定。
从单词到贺卡
ArtistAI内集成了NLP、图像生成、风格迁移等技术,3步内解决问题。
首先,我们必须要让ArtistAI的脑洞足够大,无论输入什么抽象单词,总能想到要画什么。
为此,需要使用GloVe单词嵌入模型,用维基百科词条进行训练。将输入的单词和谷歌“猜画小歌”里的图片分类进行比较,找出最贴近的分类。
不管你输入什么,ArtistAI总能联想到一个“猜画小歌”里合适的分类。
但有时它也不太聪明,比如输入美国演员“Chris Pine”,居然会联想到“树”。所以你还是对它直接一点,别太刁难了。
第二步,让ArtistAI向“猜画小歌”学会涂鸦。
在上一步中,我们让它脑洞大开想到了“树”,接着使用名为Sketch-RNN的变分自动编码器(VAE)从数据集学习足够多的涂鸦,然后ArtistAI就能画出自己的简笔画了。
最后一步,我们训练ArtistAI将涂鸦变成有名家风格的画作。
这一步中,使用了神经风格迁移技术。AI尝试理解特定画家的笔触和调色风格,并将这种风格迁移到刚才的涂鸦上。
使用教程
ArtistAI目前已经在Github上开源了,虽然叫christmAIs,为圣诞贺卡而生,但是马上新年也能用上啊。
它的安装过程非常简单。直接从GitHub上拉取项目,甚至可以傻瓜一键式安装:
source venv/bin/activate # Highly recommended
./install-christmais.sh
通过两行命令直接安装好了依赖的软件库和风格迁移模型。
然后在终端里用Python运行,命令中各项参数如下:
python -m christmais.tasks.christmais_time
--input=<输入某个单词>
--style=<你喜欢的名画所在路径>
--output=<输出图片文件路径>
这样就能获得了一张艺术风格贺卡,是不是很容易?
【来源:量子位】