如何在Python中使用OpenAI API和DALL·E 2生成图像

如何在Python中使用OpenAI API和DALL·E 2生成图像

pip install openai python-dotenv pillow

• 登录 查看 API 密钥. 如果您的 API 密钥不可见，请单击 创建新的密钥 生成一个新的。

  

• 复制生成的密钥并将其粘贴到计算机上安全的位置。 为了安全起见，您可能希望将其粘贴到项目根目录中的环境变量文件中。

如何在 Python 中使用 OpenAI API 生成和编辑图像

OpenAI API 的图像生成具有用于生成新图像、创建图像变体和编辑它们的端点。

在以下部分中，您将探索这些端点以使用自然语言提示生成、编辑、调整和下载图像。

使用 OpenAI API 生成图像

图像生成端点接受三个关键字参数。 这些包括 prompt, n, and size.

这 prompt  关键字是描述您希望生成的图像的文本字符串，而 n 是在 API 调用期间生成的图像数。 这 size尺寸 是图像大小，根据官方 OpenAI 图像生成文档，截至撰写本文时，它仅接受 256×256、512×512 和 1024×1024 像素。

以下代码使用自然语言提示生成图像并输出它们的 URL（图像地址）：

import openai
import os
import requests
from dotenv import load_dotenv
from PIL import Image
load_dotenv()

classImageGenerator:
def__init__(self) -> str:
self.image_url: str
openai.api_key = os.getenv(“OPENAI_API_KEY”)
self.APIKey = openai.api_key
self.name = None

defgenerateImage(self, Prompt, ImageCount, ImageSize):
try:
self.APIKey
response = openai.Image.create(
prompt = Prompt,
n = ImageCount,
size = ImageSize,
)
self.image_url = response[‘data’]

self.image_url = [image[“url”] for image in self.image_url]
print(self.image_url)
return self.image_url
except openai.error.OpenAIError as e:
print(e.http_status)
print(e.error)

defdownloadImage(self, names)-> None:
try:
self.name = names
for url in self.image_url:
image = requests.get(url)
for name in self.name:
with open(“{}.png”.format(name), “wb”) as f:
f.write(image.content)
except:
print(“An error occured”)
return self.name

imageGen = ImageGenerator()

imageGen.generateImage(
Prompt = “Giant lion, bear, ape, and tiger standing on a water water fall”,
ImageCount = 2,
ImageSize = ‘1024×1024’
)

imageGen.downloadImage(names=[
“Animals”,
“Animals2”
])

上面的代码包含一个 图像生成器 类与 图片_url 和 API密钥 属性。 这 生成图片 方法适应图像生成端点的要求。 它产生 n 使用自然语言提示的 URL，具体取决于 图片数 价值。

然而， self.image_url 使用列表理解中的 for 循环将生成的图像 URL 从 JSON 响应提取到列表中。

输出如下所示：

最后， 下载图片 方法通过请求生成的地址来下载每个图像。 这个方法接受 n 您希望为每个图像指定的名称数量。

使用 OpenAI API 编辑图像

编辑端点允许您根据遮罩模板编辑现有图像。 蒙版是一种 RGBA 格式，带有带注释的透明区域，并且必须与要编辑的图像大小相同。

当您提供遮罩时，编辑端点会用现有图像中的新提示替换其遮罩区域以生成新图像。 在上面的类中添加以下方法：

classImageGenerator:
    def__init__(self, ...):
        ..............

defconvertImage(self, maskName):
image = Image.open(“{}.png”.format(maskName))
rgba_image = image.convert(‘RGBA’)
rgba_image.save(“{}.png”.format(maskName))

return rgba_image

defeditImage(self, imageName, maskName, ImageCount, ImageSize, Prompt) -> str:
self.convertImage(maskName)
response = openai.Image.create_edit(
image = open(“{}.png”.format(imageName), “rb”),
mask = open(“{}.png”.format(maskName), “rb”),
prompt = Prompt,
n = ImageCount,
size = ImageSize,
)
self.image_url = response[‘data’]
self.image_url = [image[“url”] for image in self.image_url]

print(self.image_url)
return self.image_url

imageGen.editImage(
imageName = “Animals”,
maskName = “mask”,
ImageCount = 1,
ImageSize = “1024×1024”,
Prompt = “An eagle standing on the river bank drinking water with a big mountain”
)

imageGen.downloadImage(names=[
“New Animals”,
])

这 转换图像 方法将掩码图像转换为 RGBA 格式。 您可以使用 转变 来自 Python 的 pillow 包的方法（作为 PIL 导入）。

转换后，它将新蒙版保存为现有蒙版的覆盖。 因此，图像转换是编辑方法的首要任务（编辑图片) 执行。

这 图像 从现有文件中读取目标图像，而 面具 读取转换后的遮罩图像，该图像必须具有透明区域。

您可以使用 Gimp 或 Photoshop 等照片编辑软件在蒙版中标出透明区域。

例如，我们要使用上面代码中的提示编辑的图像是：

现在，假设您要按照提示中的说明将站在河岸边的羚羊换成一只老鹰。

这是面具的样子：

由于上述代码提示中的关键字是“一只站在河岸上的老鹰”，空白处被一只老鹰填满，取代了之前在那里的羚羊。

这是本例中的新图像：

您可以尝试使用其他照片来使用图像编辑端点。

使用 OpenAI API 在 Python 中生成图像变体

变体端点从现有图像生成替代图像。 通过添加图像变化方法进一步扩展图像生成器类，如下所示：

classImageGenerator:
    def__init__(self) -> str:
        .............
        
    defimageVariations(self, ImageName, VariationCount, ImageSize):
        response = openai.Image.create_variation(
            image = open("{}.png".format(ImageName), "rb"),
            n = VariationCount,
            size = ImageSize
            )
        
        self.image_url = response['data']
            
        self.image_url = [image["url"] for image in self.image_url]
        print(self.image_url)
        return self.image_url
        
imageGen = ImageGenerator()

imageGen.imageVariations(
ImageName = “New_Animals”,
VariationCount = 2,
ImageSize = “1024×1024”
)

imageGen.downloadImage(names=[
“Variation1”,
“Variation2”,
]
)

上面的代码生成图像的变体。

充分利用 OpenAI

虽然有些人担心人工智能会抢走他们的工作，但毕竟它可能是一个避风港——如果你学会控制和使用它。 这个 OpenAI 图像创建教程只是 AI 在现实世界中的众多用例之一。 OpenAI API 提供方便的预训练模型，您可以轻松将其集成到您的应用程序中。 因此，您可以接受挑战并从本教程中构建一些有价值的东西。

尽管图像生成 API 在撰写本文时仍处于测试阶段，但它已经可以让您制作想象中的艺术品。 希望它能得到进一步的更新，以允许放大并接受不同的图像尺寸。

推荐：WordPress阻止内容被复制Content Protector