如何调用百度文心 ERNIE-ViLG 的 API

文心AI作画API概述

百度文心 ERNIE-ViLG 是一款跨模态的 AI 绘画大模型，专为中文用户设计。其强大的生成能力使其在艺术创作、虚拟现实、图像编辑等领域具有广泛应用。文心 ERNIE-ViLG 的核心在于其能够根据文本描述生成图像，这一特性使其在 AI 辅助设计和虚拟数字人等领域具有广泛的应用前景。

ERNIE-ViLG生成的不同风格的图像

注册与API Key获取

在调用文心 ERNIE-ViLG 的 API 之前，首先需要在百度智能云平台上注册账户并获取 API Key 和 Secret Key。这两个密钥将用于 API 的身份验证。

注册百度智能云账号

首先，访问百度智能云官网，点击右上角的“注册”按钮。注册完成后，登录并进入控制台页面。选择“产品服务”中的“人工智能”模块，进入“智能创作平台”。

注册百度智能云账号

申请API Key和Secret Key

在“应用列表”中，选择“创建应用”，填写相关信息并选择所需的服务模块。应用创建成功后，系统将分配给您 API Key 和 Secret Key。

获取API Key

文心AI作画API接口详解

文心AI作画API主要包括两个接口：提交请求和查询结果。这两个接口的组合使用可以实现从文本到图像的生成全过程。

提交请求接口

提交请求接口用于创建图像生成任务。用户需要提供文本描述、图像分辨率、风格等参数。成功提交后，系统会返回一个任务 ID。

import requests
import json

API_KEY = "您的API Key"
SECRET_KEY = "您的Secret Key"

url = "https://aip.baidubce.com/rpc/2.0/ernievilg/v1/txt2img?access_token=" + get_access_token()
payload = json.dumps({
    "text": "中国山水画",
    "resolution": "1024*1024",
    "style": "古风",
    "num": 2
})
headers = {
    'Content-Type': 'application/json',
    'Accept': 'application/json'
}
response = requests.request("POST", url, headers=headers, data=payload)
print(response.text)

查询结果接口

查询结果接口用于检查任务状态并获取生成的图像 URL。用户需要提供任务 ID 来查询生成状态。

import requests
import json
from io import BytesIO
from PIL import Image
import os

API_KEY = "您的API Key"
SECRET_KEY = "您的Secret Key"

def main():
    url = "https://aip.baidubce.com/rpc/2.0/ernievilg/v1/getImg?access_token=" + get_access_token()
    payload = json.dumps({
        "taskId": "您的任务ID"
    })
    headers = {
        'Content-Type': 'application/json',
        'Accept': 'application/json'
    }
    response = requests.request("POST", url, headers=headers, data=payload)
    print('请求的返回值：', response.text)

    json_result = json.loads(response.text)
    imgUrls = json_result['data']['imgUrls']
    for i, img_url in enumerate(imgUrls):
        req = requests.get(img_url['image'])
        image = Image.open(BytesIO(req.content))
        image.save(os.path.join('图片保存路径', f'{i}.jpg'), 'JPEG')

代码实现指南

在实现调用百度文心 ERNIE-ViLG 的 API 时，首先需要确保本地开发环境配置正确。以下是实现代码调用的步骤和注意事项。

环境配置要求

Python：需要 3.7 及以上版本。
Pycharm：推荐使用最新版以避免兼容性问题。
Conda：用于创建和管理虚拟环境。

代码示例

以下代码展示了如何使用 Python 调用文心 AI 作画 API。确保在运行代码前，已正确安装 requests 和 PIL 库。

import requests
import json

API_KEY = "您的API Key"
SECRET_KEY = "您的Secret Key"

def get_access_token():
    url = "https://aip.baidubce.com/oauth/2.0/token"
    params = {"grant_type": "client_credentials", "client_id": API_KEY, "client_secret": SECRET_KEY}
    return str(requests.post(url, params=params).json().get("access_token"))

if __name__ == '__main__':
    main()