通过文字描述一个场景或一个人,然后计算机就会按照你的描述生成画作。目前有很多开源或闭源产品,比如最近比较火的Stable Diffusion 、DALL·E,国内的Paddlepaddle都实现从文本到图像。
但是,在生成作品的质量上,Stable Diffusion是最好的。今天先来分享一下,使用Paddlepaddle如何实现从文字到图像。
首先,我们需要按照Python开发环境,然后安装下面的两个包:
pip install paddlepaddle==2.3.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install paddlehub -i https://pypi.tuna.tsinghua.edu.cn/simple
接下来就可以编写具体,文本生成图像的代码了。
import paddlehub as hub
module = hub.Module(name="ernie_vilg")
results = module.generate_image(text_prompts=["虎皮鹦鹉"])
第三行代码,使用机器学习模型ernie_vilg,text_prompts参数是文字部分,比如这里我们输入的是"虎皮鹦鹉",就这么简单,三行代码即可完成。
虎皮鹦鹉
可爱的孩子
以上两幅作品就是我通过Paddlepaddles'h生成的,肆意发挥你的创意吧。持续关注TalkPythonji解锁更多技能。