OpenAI今年1月公布的Dall-E可谓艳惊四座,只需输入一段话,就能根据内容输出一幅画。
连吴恩达老师都不禁点赞。

比如让它设计一个“像牛油果一样的座椅”。

但是,OpenAI只在官网设置了几种固定模板,代码也没有开源,用户无法用AI设计出更多的图片。

不过现在有个好,一位华人小哥Philip Wang已经复现了一个PyTorch版Dall-E,叫做Deep Daze。

请给我一个五彩斑斓的黑只需一行敕令就能让AI画画

他利用了OpenAI的CLIP以及斯坦福大学的Siren,可以在个人电脑上运行。

只需输入在命令行中输入一段描述,这个AI就能画出各式图片。

比如在绿色山丘上的薄雾(mist over green hills):

或者是“草地上的碎盘子”(shattered plates on the grass):

你以为实际效果如何?

开拓该项目的Philip Wang,便是在今年年初公布要复现Dall-E的那位大神,没想到不到3个月的韶光,这个项目已经如此强大,在GitHub上已收成1.1k星。

安装与利用

Deep Daze可直接通过pip安装:

pip install deep-daze

安装完成后,输入以下命令运行。

imagineTEXT<flags>

个中imagine是命令,TEXT必须是一段不超过77个字符的英文描述,<flags>为可选参数,可加可不加。

如果我们想天生文章开始那张山中薄雾的图片,就输入以下命令:

imagine mist over green hills

<flags>中可加入的参数有:—deeper,获取更高的图像质量;—num-layers,指定网络的层数……更多用法可以去GitHub项目主页查询。

如果担心AI的想象力过于天马行空,可以指定一张图片,让AI在此图像根本上想象。
这样就能担保图像不至于太过离谱。

详细的命令如下:

imagine ‘description’ —start-image-path /path/to/picture.jpg

假设我们给AI一张披萨的图片,让它想象“上面铺满青椒的样子”。

那么我们将上述命令中的描述改成“A pizza with green pepper”。

你以为这便是Deep Daze的全部?实在,它还有一个更强大的完备体——故事模式。

只需设置—create_story=True,就可以解锁该模式,取消77个字符的限定,项目作者将美国墨客罗伯特·弗罗斯特的诗《雪夜林边歇马》输入个中,得到了一段mp4视频。

到这里我有点想让AI画一下“五彩斑斓的黑”,但贫穷限定了我的操作。

由于利用Deep Daze必须要安装CUDA,而且最好在不小于8GB显存的GPU上,才能得到不错的效果。
而我现在根本买不起英伟达显卡。

项目地址:https://github.com/lucidrains/deep-daze

— 完 —

量子位 QbitAI · 头条号签约

关注我们,第一韶光获知前沿科技动态