ERNIE-Image Atlas Cloud API + 批量生产：企业级 AI 绘画管线
百度 Atlas Cloud 提供 ERNIE-Image API 服务——本文教你搭建企业级批量图像生成管线。
一、Atlas Cloud 平台介绍
百度智能云 Atlas 平台提供 ERNIE-Image 的云端 API 服务，无需本地 GPU 即可调用。
核心特性
| 特性 | 说明 |
|---|---|
| 按需付费 | 按生成数量计费，无初始投入 |
| 弹性扩展 | 支持高并发批量生成 |
| 全球加速 | 全球节点部署，低延迟 |
| SLA 保障 | 99.9% 可用性 |
访问入口
二、API 调用基础
获取 API Key
- 登录百度智能云控制台
- 创建应用
- 获取 Access Key 和 Secret Key
Python 调用示例
import requests
import base64
def ernie_image_api(prompt, access_key, secret_key):
url = "https://wenxin.baidu.com/v1/images/generations"
headers = {
"Authorization": f"Bearer {get_token(access_key, secret_key)}",
"Content-Type": "application/json"
}
data = {
"model": "ernie-image-8b",
"prompt": prompt,
"n": 1,
"size": "1024x1024",
"response_format": "b64_json"
}
response = requests.post(url, headers=headers, json=data)
result = response.json()
image_data = base64.b64decode(result["data"][0]["b64_json"])
return image_data
def get_token(access_key, secret_key):
url = f"https://aip.baidubce.com/oauth/2.0/token"
params = {
"grant_type": "client_credentials",
"client_id": access_key,
"client_secret": secret_key
}
return requests.get(url, params=params).json()["access_token"]
三、批量生产管线
架构设计
CSV/JSON 输入 → 任务队列 → API 调用 → 图像存储 → 结果输出
批量生成脚本
import csv
import requests
import base64
import time
from concurrent.futures import ThreadPoolExecutor
def batch_generate(csv_file, access_key, secret_key, output_dir="./output"):
# 读取 CSV
with open(csv_file, 'r', encoding='utf-8') as f:
reader = csv.DictReader(f)
tasks = list(reader)
# 批量生成
def generate_task(task):
image_data = ernie_image_api(
task['prompt'],
access_key,
secret_key
)
# 保存图像
import os
os.makedirs(output_dir, exist_ok=True)
filename = f"{task['id']}.png"
with open(f"{output_dir}/{filename}", 'wb') as f:
f.write(image_data)
return task['id']
# 并发执行
with ThreadPoolExecutor(max_workers=10) as executor:
results = executor.map(generate_task, tasks)
print(f"Generated {len(list(results))} images")
CSV 输入模板
id,prompt,negative_prompt,width,height
1,"a cat on a table","","1024","1024"
2,"a dog in a garden","","1024","1024"
3,"a bird in a tree","","1024","1024"
四、ComfyUI + API 混合管线
架构
本地 ComfyUI（复杂任务） ↔ Atlas API（简单任务）
任务分流逻辑
| 任务类型 | 处理位置 | 原因 |
|---|---|---|
| 简单生成 | Atlas API | 成本低，速度快 |
| IP-Adapter | 本地 ComfyUI | API 不支持 |
| ControlNet | 本地 ComfyUI | API 不支持 |
| Inpainting | 本地 ComfyUI | 精细控制 |
| 批量生产 | Atlas API | 弹性扩展 |
五、企业级部署方案
方案 1：全云端
优势：
- 零硬件投入
- 弹性扩展
- 维护成本低
成本：
- API 调用：0.01 元/张
- 存储：0.1 元/GB/月
方案 2：混合云
优势：
- 核心数据本地
- 弹性扩展云端
- 成本控制灵活
成本：
- 本地 GPU：5-10 万
- API 调用：按量
六、总结
Atlas Cloud API + 批量生产核心要点：
- API 调用：简单集成，按需付费
- 批量管线：CSV/JSON 驱动 + 并发执行
- 混合部署：复杂任务本地，简单任务云端
- 企业级：SLA 保障 + 全球加速
掌握 Atlas Cloud API，ERNIE-Image 成为你的企业级 AI 绘画引擎。
本文基于百度智能云 Atlas + ERNIE-Image API。