volc-image
火山引擎图像生成技能 - 使用火山引擎方舟(Doubao Seedream) API生成图片。
环境配置
需要设置环境变量 ARK_API_KEY:
功能
- 生成图片 - 根据提示词生成图片
- 下载保存 - 自动将生成的图片保存到本地
使用方式
直接告诉我要生成什么样的图片,例如:
- "生成一张小说封面:顶流重生指南"
- "生成一张科技感海报"
- "生成一张可爱猫咪图片"
参数说明
| 参数 | 说明 | 默认值 |
|---|
| model | 模型 | doubao-seedream-5-0-260128 |
| size | 尺寸 | 2K (1664x2496) |
| prompt | 提示词 | 用户输入 |
示例提示词
小说封面
小说封面,顶流重生指南,标题金色大字,时尚美女坐在王座上,娱乐圈背景,霓虹灯光,紫色金色主题,精致画面,4K
科幻风格
星际穿越,黑洞,黑洞里冲出一辆支离破碎的复古列车,视觉冲击力,电影大片,末日既视感,动感,对比色,oc渲染,光线追踪,动态模糊,景深,超现实主义,深蓝
可爱风格
可爱的小猫咪,胖乎乎,大眼睛,萌萌的,粉色背景,卡通风格,精致可爱
技术实现
import os
import requests
# API配置
url = 'https://ark.cn-beijing.volces.com/api/v3/images/generations'
api_key = os.environ.get('ARK_API_KEY')
# 请求数据
data = {
'model': 'doubao-seedream-5-0-260128',
'prompt': '<用户提示词>',
'sequential_image_generation': 'disabled',
'response_format': 'url',
'size': '2K',
'stream': False,
'watermark': True
}
# 发送请求
response = requests.post(url, headers={
'Authorization': f'Bearer {api_key}',
'Content-Type': 'application/json'
}, json=data)
# 获取图片URL并下载
img_url = response.json()['data'][0]['url']
img_data = requests.get(img_url).content
# 保存图片
with open('output.jpg', 'wb') as f:
f.write(img_data)
注意事项
- API Key 需要在火山引擎控制台获取
- 图片URL有效期约24小时
- 生成图片可能需要几秒钟时间
执行命令
设置API Key并生成图片:
# 设置API Key
set ARK_API_KEY=你的API密钥
# 生成图片
python volc_image.py "你的提示词"
或者在Python中直接调用:
import os
os.environ['ARK_API_KEY'] = '你的API密钥'
exec(open('volc_image.py').read())