AI 教你画油画:任意画风都可驾驭,笔画序列秒秒钟呈现

2021/8/16 10:37:27 来源:量子位 作者:杨净 责编:骑士

AI 已经能教你画油画了。

随便给一张图,笔画序列秒秒钟呈现。

比如世界名画蒙娜丽莎。

亦或是写实类的小鸟。

还有极具氛围感的河灯。

总之什么风格都可以驾驭。

这项技术在 Reddit 上 21 小时内就已经有 600 + 的点赞量。

究竟是如何打造的呢?

用前馈网络预测笔画

神经绘画,就是为给定的图像生成一系列笔画,并使用神经网络进行绘画式的真实再现过程。

研究团队提出了一个基于 Transformer 的框架,叫做 Paint Transformer,用前馈网络来预测笔画的参数。

由于当前没有可用的数据集来训练 Paint Transformer,受物体检测启发,研究人员设计了一个自训练 Pipeline。

整个模型由两个模块组成:笔画预测器和笔画渲染器。

给定目标图像和中间画布图像,笔画预测器,生成一组参数以确定当前笔画集。

预测器包含了两个用于特征嵌入的 CNN 网路和一个用于参数预测的 Transformer。

随后,笔画渲染器为笔画集汇总的每个笔画,生成笔画图像,并将它们绘制到画,产生结果图像,大小为 512*512。

在 DETR(用 Transformer 进行对象检测)的基础上,增加了二进制神经元来预测笔画是否应该被保留。

这样它就可以在没有任何现成的数据集的情况下进行训练,同时还能实现出色的泛化能力。

实验表明,这一方法比以前的方法取得了更好的绘画性能,而且训练和推理成本更低。

百度南大团队打造

这项技术由百度、南京大学、罗格斯大学共同打造。

目前代码已经开源,并在一刻相册 App 上应用。

论文地址:

https://arxiv.org/abs/2108.03798

GitHub 网址:

https://github.com/wzmsltw/PaintTransformer

软媒旗下网站: IT之家 辣品 - 超值导购,优惠券 最会买 - 返利返现优惠券 iPhone之家 Win7之家 Win10之家 Win11之家

软媒旗下软件: 魔方 旗鱼浏览器(极速内核) 云日历 酷点桌面 Win7优化大师 Win10优化大师 软媒手机APP应用