使用火山引擎豆包 doubao-seedream-5-0-260128 模型,根据文字描述生成高质量图像,支持并发批量生成多张图片,并输出图库预览页面。
pip install "openai>=1.0"API Key 读取优先级(从高到低):
--api-key 命令行参数ARK_API_KEY~/.doubao-image-gen/.env 文件中的 ARK_API_KEY=xxx获取 API Key:登录 火山方舟控制台 → API Key 管理
bash
# 生成单张图片
python {baseDir}/scripts/gen.py --prompt "赛博朋克风格的上海夜景" --api-key YOUR_KEY
# 并发生成4张(默认并发数=4)
python {baseDir}/scripts/gen.py --prompt "水墨风格的山水画" --count 4 --api-key YOUR_KEY
# 指定尺寸(支持 1024x1024 / 2K / 1280x720 / 720x1280 / 2048x2048)
python {baseDir}/scripts/gen.py --prompt "星空下的草原" --size 2K --api-key YOUR_KEY
# 指定输出目录
python {baseDir}/scripts/gen.py --prompt "古风仙侠" --out-dir ./output --api-key YOUR_KEY
# 从环境变量读取 Key(推荐)
python {baseDir}/scripts/gen.py --prompt "未来城市" --count 2
参数默认值说明--prompt必填图像描述提示词--count1生成数量(并发执行)--size``2K图像尺寸--model``doubao-seedream-5-0-260128模型名称--out-dir``./doubao-output-{时间戳}输出目录--api-key环境变量ARK API Key--workers4并发线程数--watermarkFalse是否添加水印--dry-runFalse仅打印参数不调用 API
*.jpeg 图像文件(按序号命名)prompts.json 提示词与文件的映射记录index.html 图库预览页面(可直接在浏览器打开)当用户说以下内容时,加载本技能并调用脚本:
标准流程:
python {baseDir}/scripts/gen.py 生成图片index.html 预览链接供浏览示例 Prompt 优化: 用户说"画一只猫"→ 优化为 "A cute cat sitting gracefully, soft studio lighting, photorealistic, 8K detail"