中国风？古典系？AI中文绘图创作尝鲜！⛵

本文对百度开源的中文 AI 图像生成器 ERNIE 进行讲解，包括工具简介、通过Demo页面和API两种使用方式，以及使用中文提示词生成的图像效果。

---

作者：韩信子@ShowMeAI
深度学习实战系列：https://www.showmeai.tech/tutorials/42
本文地址：https://www.showmeai.tech/article-detail/413
声明：版权所有，转载请联系平台与作者并注明出处
收藏ShowMeAI查看更多精彩内容

引言

人工智能图像生成器现在风靡一时。虽然 Dall-E2、 MidJourney和 Stable Diffusion目前处于中心舞台，但是百度开发的开源 AI 图像生成器 ERNIE-ViLG 也在互联网上流传。

ERNIE-ViLG 是什么

ERNIE 代表通过知识整合增强表示——视觉语言生成。它是一个统一的生成预训练框架，用于使用 Transformer 模型进行双向图像文本生成。

需要明确的是，ERNIE 并不是 Stable Duffusion 的一个变体版本，尽管这两种模型都是开源的。根据 ERNIE论文，ERNIE-ViLG 2.0 是一种改进的文本到图像扩散模型，具有知识增强的去噪专家组合，以结合有关视觉场景的额外知识，并在不同步骤中解耦去噪能力。

ERNIE-ViLG 使用方法

现在可以通过 huggingface 免费访问该模型。我们可以在 ERNIE-ViLG Demo 查看工具实例。

界面非常简单:可以看到经典的提示框，各种艺术风格的选择，以及图像分辨率。在艺术风格方面，ERNIE 提供了一些有趣的选择。这让我想起了 Dream by WOMBO 中的一个类似功能。

让我们用提示“An astronaut riding a horse in space”来生成样本。

该应用程序生成六张分辨率为 1024x1024 的图像。

ERNIE 的研究人员声称，他们的方法比 Dall-E2 和 Stable Diffusion 产生更清晰、更自然的细节。

真的吗？现在让我们尝试将艺术风格更改为“油画”。

这些结果本身都非常惊艳。

质量上，ERNIE 可以与 Dall-E2 和 Stable Diffusion 相媲美。但是，它在一致性方面表现不佳，就像在上面提到的，其中两个结果中的马无处可寻。这可能是训练ERNIE的数据集较小所致。

不过，如果仔细考虑数据集的大小，训练了 14.5 亿张图像的 ERNIE 与 MidJourney、Google Parti 等大型竞争对手的差距并不大。

也许开发人员找到了一种优化算法的方法，使其能够在较小的数据集上运行良好。我假设如果模型是在更大的数据集上训练的，现有的限制就会消失。

需要特别注意的是，ERNIE 是用中文开发的。这意味着，在开始图像生成过程之前，必须将英文提示翻译成中文。当然，使用中文的宝宝可以尽情玩耍~

这里有更多示例结果。

上图提示文字：“戴眼镜的猫”风格“油画”

上图提示文字：“五官端正的绘画女生”

API 访问 ERNIE-ViLG

您可以通过 API 访问 ERNIE。如果您想尝试使用 API，请按照这个 GitHub 上的指南进行操作。

示例 API 调用如下所示。

def generate_image(
          text_prompts:str,
          style: Optional[str] = "探索无限",
          topk: Optional[int] = 6,
          output_dir: Optional[str] = 'ernievilg_output')

• text_prompts：输入提示
• style ：生成图像的风格
• topk：生成图像的数量（最多 6 个）
• output_dir：保存输出图像的目录

ERNIE 的使用指南还有一个建议列表，可帮助您找到结合使用样式和修饰符的最佳方式。

• 【作图规则】Prompt构建是文本符合逻辑的组合，有序且丰富的描述可以不断提升画面效果
• 【新手入门】不知如何输入Prompt？点击示例，体验文生图的魅力，参考教程，逐步进阶~
• 【风格生成】试试添加 “国潮”、“国风”等，感受中国风的魅力
• 【风格生成】试试混合两种代表性的风格，例如“赛博朋克，扁平化设计”、”皮克斯动画，赛博朋克”
• 【人像生成】添加“仙鹤、月亮、楼阁、小屋、街道、玫瑰、机械”，画面会更饱满
• 【人像生成】添加“精致面容、唯美、cg感、细节清晰“等，人物刻画会更细致
• 【风格生成】添加“扁平化风格，logo”等，可以设计出各类图标等，例如 “猫猫头像，扁平化风格”
• 【风格生成】指定颜色，或添加“烟雾缭绕”、“火焰”、“烟尘”、“花瓣”，生成画面的氛围感更饱满
• 【创意生成】发挥想象力，例如：“中西混搭”、“泰迪熊唱京剧”、“米老鼠吃火锅”
• 【风格生成】“水彩”，“水墨”与古诗组合，画面意境会有提升~
• 【风格生成】想要日系头像和拟人化动物？试试关键词“日系手绘”、“治愈风”
• 【风格生成】添加“pixiv”，生成二次元或者动漫的画质更惊艳

参考资料

• Dall-E2
• MidJourney
• Stable Diffusion
• ERNIE-ViLG
• ERNIE论文
• ERNIE-ViLG Demo
• Dream by WOMBO
• PaddleHub ERNIE-ViLG GitHub

推荐阅读

数据分析实战系列：https://www.showmeai.tech/tutorials/40

机器学习数据分析实战系列：https://www.showmeai.tech/tutorials/41

深度学习数据分析实战系列：https://www.showmeai.tech/tutorials/42

TensorFlow数据分析实战系列：https://www.showmeai.tech/tutorials/43

PyTorch数据分析实战系列：https://www.showmeai.tech/tutorials/44

NLP实战数据分析实战系列：https://www.showmeai.tech/tutorials/45

CV实战数据分析实战系列：https://www.showmeai.tech/tutorials/46

AI 面试题库系列：https://www.showmeai.tech/tutorials/48