[文心大模型创意项目]还在愁没有头像？属于你的个人头像来了！

最近抖音火起来了职业头像这个玩意，这里，我们来用文心大模型来生成！代码无脑一键运行，生成属于你的独一无二的头像！版本2是hub调用版，版本3是wenxin-api调用版。

AI Studio

885人浏览 · 2022-09-03 17:12:13

AI Studio · 2022-09-03 17:12:13 发布

还在愁没有头像？属于你的个人头像来了！

（！！！别走开！文中有奖！！！）

版本2为hub版本，全代码无脑运行

版本3为wenxin-api版本，需要去申请ak、sk，但更稳定一些.

在这里插入图片描述

最近抖音又火起来了一个职业装头像，将其作为微信的背景和头像就很nice。

如下：（图源抖音，侵删）
在这里插入图片描述

这里

我们可以发挥自己的想象力

挖掘你的脑洞里面合适的提示词->Prompt

生成我们想要的风格的头像

当然，最新的文心大模型的调用可以用paddlehub直接调用，也是非常的方便。

PS

prompt 是提示词的意思，在我们这里，就是指，你输入的文字是一个稳定的关键词，可以在反复使用中都保持一个不错的、稳定的效果。

比如：文心目前已知的[像素风、赛博朋克、玻璃艺术、混乱艺术]就是不错的关键词。

看这里看这里最佳头像征集

这里大家如果生成了特别好看的或者特别的头像以及不错的prompt可以放到评论区。

项目维持到2022年9月14日12:00，大家可以去点赞，点赞最高的我个人给出66.66的红包

评论可以获奖的要求是：

fork本项目
评论区评论：图像加生成图像的prompt（我这里不做限制）。

我这里给大家打个样！

在这里插入图片描述

(你们可以只放上面一排的图即可)

2022.08.30 目前收到的比较不错的效果图：

在这里插入图片描述

当然除此之外，文心大模型最近还有活动：

如果你找到了一个特别好用的风格的prompt，可以将其写成一个小项目展示你的prompt的稳定性。

GT WeChat： GT_ZhangAcer

在这里插入图片描述

1 pip安装wenxin

这个项目就不再详细的描述ernie-vilg了

细节可以去看下面链接的-5-版本

在这里插入图片描述

这个七夕不太冷~你的专属AI女友给你唱爱你

!pip install wenxin-api

2 输入你的prompt

我这里就简单做个演示，以及我找到的不错的头像~

AK和SK的获取：

https://wenxin.baidu.com/moduleApi/key

# 你的prompt 我这里发现用逗号连接词语效果更佳
# text_prompt = "程序员，写代码"
text_prompt = "程序员在写代码，科幻，头像艺术，像素艺术"

# -*- coding: utf-8 -*
import wenxin_api # 可以通过"pip install wenxin-api"命令安装
from wenxin_api.tasks.text_to_image import TextToImage
wenxin_api.ak = "your ak"
wenxin_api.sk = "your sk"
input_dict = {
    "text": text_prompt,
    "style": "油画"
}
rst = TextToImage.create(**input_dict)

print(rst)

2022-08-28 12:38:13,916 - model is painting now!, taskId: 1062464, waiting: 30s
2022-08-28 12:38:34,290 - model is painting now!, taskId: 1062464, waiting: 30s
{'imgUrls': ['https://wenxin.baidu.com/younger/file/ERNIE-ViLG/4951a8e94cf3461f8e30e3f25f9d07b9ex', 'https://wenxin.baidu.com/younger/file/ERNIE-ViLG/4951a8e94cf3461f8e30e3f25f9d07b9i4', 'https://wenxin.baidu.com/younger/file/ERNIE-ViLG/4951a8e94cf3461f8e30e3f25f9d07b95q', 'https://wenxin.baidu.com/younger/file/ERNIE-ViLG/4951a8e94cf3461f8e30e3f25f9d07b930', 'https://wenxin.baidu.com/younger/file/ERNIE-ViLG/4951a8e94cf3461f8e30e3f25f9d07b9v9', 'https://wenxin.baidu.com/younger/file/ERNIE-ViLG/4951a8e94cf3461f8e30e3f25f9d07b9a2']}

import requests
import os
from PIL import Image
import matplotlib.pyplot as plt
%matplotlib inline

# 生成存图的目录
def init_mkdir():
    data_path = 'output-img'
    if not os.path.exists(data_path): # 判断文件夹是否存在
        os.makedirs(data_path) # 不存在则新建文件夹

init_mkdir()

# 展示生成的图片并保存
for i in rst['imgUrls']:
    #使用requests直接get 下载图片使用
    r = requests.get(i)
    ii = i.split('/')[-1]+'.png'
    out_ii = 'output-img/' + ii
    with open(out_ii, 'wb') as f:
        f.write(r.content)
    f.close()
    img_ = Image.open(out_ii)
    plt.show()
    plt.imshow(img_)