如果我想批量生成图片,并且把生成的图片同一保存在电脑D:\研一下\视觉框架\全球南方\AI相关\out test文件夹中,以上代码应该如何修改?
时间: 2024-10-16 07:22:33 浏览: 28
如果你想批量生成图片并保存到指定目录,你可以对上述代码做一些调整。首先,你需要创建一个包含所有你要生成图像的提示列表,然后循环遍历这个列表,每次生成一张图片并保存。这里假设你使用的是一些支持生成图片的API,并且能够处理JSON格式的数据。
```python
import os
import requests
# 假设这是一个提示列表
prompts = ["image prompt 1", "image prompt 2", ...]
# 指定保存图片的路径
save_path = r"D:\研一下\视觉框架\全球南方\AI相关\out test"
# API调用函数
def generate_image(prompt):
api_url = "https://your-api-url.com/stable-diffusion/generate-image"
# ... (其他API设置)
try:
# 发送POST请求生成图片
response = requests.post(api_url, json={"prompt": prompt})
if response.status_code == 200:
# 解析返回的图片数据
image_data = response.json()
filename = f"{prompt}.jpg" # 这里假设返回的是JPEG格式的图片
full_path = os.path.join(save_path, filename)
# 保存图片
with open(full_path, 'wb') as file:
file.write(image_data['image'])
print(f"Image saved: {full_path}")
else:
print(f"Error: {response.status_code} - Saving failed")
except Exception as e:
print(f"Error saving image: {e}")
# 批量生成并保存图片
for prompt in prompts:
generate_image(prompt)
```
在这个示例中,每张图片会被命名为提示内容加上".jpg"后缀,并保存在指定的"out test"文件夹内。
阅读全文