对于绘画一窍不通的我,也喜欢看动漫,看到一些绝美的画,何尝不会感叹,要是我也会画画就好了,现在终于有机会帮我实现这个想法了,无意间看到了AI的一个新应用,最近热度还挺高的AI绘画,充分引起了我的兴趣,于是乎我决定试用市面上主流的几款,包括了ERNIE-ViLG,DALL·E,Imagen,Midjourney,Disco Diffusion 从安装到简单的使用,接下来让我们先来看看支持中文的文转图模型ERNIE-ViLG,到底有多么好玩吧。
-
安装环境如下
win10,python3.8
-
ERNIE-ViLG介绍
文心ERNIE-ViLG是百度开发的,参数规模达到100亿,是目前为止全球最大规模中文跨模态生成模型,可以输入一段文本描述以及生成风格,模型就会根据输入的内容自动创作出符合要求的图像。github地址
https://github.com/PaddlePaddle/PaddleHub/tree/develop/modules/image/text_to_image/ernie_vilg
-
安装开源预训练模型版本过程
根据步骤进行安装,我用的是cpu版本,如果系统环境变量pip路径是python3.8下的pip可以直接使用下面命令安装
安装paddlepaddle,版本需要大于2.0.0,直接安装的版本满足
pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple/
安装paddlehub, 版本需要大于2.2.0,直接安装的版本满足
pip install paddlehub -i https://mirror.baidu.com/pypi/simple/
安装paddlepaddle的ernie_vilg模块
hub install ernie_vilg
-
测试
安装完成预训练模型以后,可以直接使用下面的代码进行简单测试,结果会生成6幅油画。
import paddlehub as hubmodule = hub.Module(name="ernie_vilg")text_prompts = ["炎热的夏日,蓝天白云,公路边上一望无垠金黄色的稻田,戴着草帽的农夫在稻田里收割水稻"]#参数介绍,text_prompts是要画的内容描述,style是画的风格,output_dir是结果的输出目录,里面最为关键的是text_prompts参数images = module.generate_image(text_prompts=text_prompts, style='油画',output_dir='./ernie_vilg_out/')
根据输出来看,5幅画中都没有农民在稻田里收割的细节,不过仅仅这么一句话,就能出来这个效果感觉还不错。
简单的介绍就到这里了,如果大家还想具体知道如何画出更好更美的可以点击下面的链接查看具体内容。
有了AI智能绘画,我也可以成为绘画大师——全球最大规模中文跨模态生成模型ERNIE-ViLG
如果大家喜欢也可以关注下面的微信公众号PythonerLeague,会有更多优质python技术文章发布,让我们不一起进步,有问题也可以提问,互相讨论帮助