用Openai的dall-e2生成图片的代码

使用DALL-E 2生成图片需要先安装OpenAI API和相关的Python库。以下是生成图片的基本代码:

import openai
import requests
from requests.structures import CaseInsensitiveDict

import json

openai.api_key = "YOUR_API_KEY"

image_prompt = "a portrait of a cat"
openai_model = "image-alpha-001"
image_size = 256

def generate_image(prompt, model, size):
    
    headers = CaseInsensitiveDict()
    headers["Content-Type"] = "application/json"
    headers["Authorization"] = f"Bearer {openai.api_key}"

    data = """
    {
        """
    data += f'"model": "{model}",'
    data += f'"prompt": "{prompt}",'
    data += f'"num_images":1,'
    data += f'"size":"{size}"'
    data += """
    }
    """

    resp = requests.post("https://api.openai.com/v1/images/generations", headers=headers, data=data)

    if resp.status_code != 200:
        raise ValueError("Failed to generate image")

    response_text = json.loads(resp.text)
    image_url = response_text['data'][0]['url']

    return image_url

image_url = generate_image(image_prompt, openai_model, image_size)
print("Image URL:", image_url)

这里使用了OpenAI API的“images”功能,通过post请求将生成图片的参数传递给API,获取url链接后,你可以将生成图片的代码进一步完善,实现更多自定义参数的功能,例如生成多张图片、调整图片大小、添加图像元素等。

猜你喜欢

转载自blog.csdn.net/zjj1898/article/details/129870534