Python| 如何使用 DALL·E 和 OpenAI API 生成图像(1)

数据科学工厂

引言

想象一下，只要描述你想要的画面，电脑就能帮你画出来。这在几年前还像是科幻小说里的场景，但随着神经网络和潜在扩散模型（LDM）技术的发展，现在已经成为可能。OpenAI 推出的 DALL·E 工具，因其能生成令人惊叹的艺术作品和逼真的图像而广受欢迎。

你可以通过 OpenAI 的 API 访问 DALL·E，这样你就可以将它的功能集成到你的 Python 程序中。

本教程将带你：

• 快速上手 OpenAI 的 Python 库
• 探索与图像生成相关的 API 调用
• 根据文本提示生成图像
• 制作生成图像的不同版本
• 将 Base64 格式的 JSON 响应转换成 PNG 图像文件

环境准备

如果你对 DALL·E 的强大功能感到好奇，并希望将其集成到你的 Python 项目中，那么这篇文章正是你需要的。在本节中，你将快速了解如何开始在你的代码中利用 DALL·E 来创建图像。

• 安装 OpenAI Python 库

请确保你的 Python 版本至少是 3.7.1，并创建一个虚拟环境来隔离项目依赖。接下来，安装 OpenAI 提供的 Python 库，以便使用 DALL·E 的功能。

$ python --version
Python 3.12.5
$ python -m venv venv
$ source venv/bin/activate
(venv) $ python -m pip install openai

openai 包让您可以访问完整的 OpenAI API。在本教程中，您将重点关注图像生成，它使您可以与 DALL·E 模型交互，根据文本提示创建和编辑图像。

• 获取您的 OpenAI API 密钥

您需要 API 密钥才能成功进行 API 调用。注册 OpenAI 并通过单击仪表板菜单，然后单击导航菜单左下角的 API 密钥来创建新的项目 API 密钥：

在此页面上，您可以创建和管理您的 API 密钥，这使您可以访问 OpenAI 通过其 API 提供的服务。您可以创建和删除密钥。单击“创建新密钥”以创建新的 API 密钥，并复制弹出窗口中显示的值。复制此键的值，以便稍后可以在项目中使用它。您只会看到该键值一次。

• 将您的 API 密钥保存为环境变量

保存 API 密钥并使其可用于 Python 脚本的一种快速方法是将其保存为环境变量。选择您的操作系统以了解如何：

(venv) $ export OPENAI_API_KEY="<your-key-value-here>"

通过执行这条指令，你可以在当前的终端会话中设置一个名为 OPENAI_API_KEY 的环境变量，以便访问你的 API 密钥。请记住，一旦你关闭了终端窗口，这个密钥就会失效。

你可以根据个人喜好来命名这个环境变量，但如果你按照 OpenAI 的官方文档建议，使用 OPENAI_API_KEY 这个名字，那么你就能够直接使用文档中的代码示例，无需进行额外的配置。

现在，准备工作已经完成，你的 API 密钥也已安全存储，你可以开始根据文本提示来生成图像了。

根据文本描述生成图像

首先，通过命令行界面调用 openai 库来确保你的环境已经配置完毕，准备开始使用。

(venv) $ openai api images.generate -p "a vaporwave computer"

该命令将向 OpenAI 的图像 API 发送请求，并根据文本提示“蒸汽波计算机”创建图像。因此，您将收到一个 JSON 响应，其中包含指向您新创建的图像的 URL：

{
  "created": 1723549436,
  "data": [
    {
      "b64_json": null,
      "revised_prompt": null,
      "url": "https://oaidalleapiprodscus.blob.core.windows.net/private/org
      ⮑ -H8Satq5hV2dscyutIStbN89m/user-O5O8mpE4TugEI9zqpxx8tGPt/img-4rDZEQ
      ⮑ leCdzAD3fuWowgLb5U.png?st=2024-08-13T10%3A43%3A56Z&se=2024-08-13T1
      ⮑ 2%3A43%3A56Z&sp=r&sv=2024-08-04&sr=b&rscd=inline&rsct=image/png&sk
      ⮑ oid=d505667d-d6c1-4a0a-bac7-5c84a87759f8&sktid=a48cca56-e6da-484e-
      ⮑ a814-9c849652bcb3&skt=2024-08-12T23%3A00%3A04Z&ske=2024-08-13T23%3
      ⮑ A00%3A04Z&sks=b&skv=2024-08-04&sig=c%2BBmroDkC0wkA2Lu8MxkvXH4xjdW5
      ⮑ B18iQ4e4BwJU6k%3D"
    }
  ]
}

单击您的 URL 或将其复制并粘贴到浏览器中以查看图像。这是 DALL·E 针对这一要求设想的图像：

你的生成图像会是独一无二的。这是因为扩散模型仅在你发起请求时才生成图像。

提示：你生成的图像的链接仅在一小时内有效，所以如果你对生成的图像感到满意并希望长期保存，请务必及时将其下载到你的电脑上。

当你向 API 发送请求时，需要遵守 OpenAI 的使用条款。如果你提供的文本提示违反了这些条款，你将收到错误提示，并且如果屡次违规，你的账户可能会被封禁。

既然你已经验证了所有设置都正确无误，并且对使用 OpenAI API 生成图像的功能有了初步了解，接下来你将学习如何将其整合到 Python 脚本中。

调用 API

能够通过命令行界面（CLI）生成图像固然不错，但如果能将这项功能集成到你的 Python 应用中就更棒了。你可以借此构建许多激动人心的项目！

打开你常用的代码编辑器，编写一个脚本。这个脚本将用于根据文本提示生成图像，操作方式与你之前在命令行中进行的操作类似。

但这一次，你将明确指定要使用的 DALL·E 模型版本：

from openai import OpenAI

client = OpenAI()

PROMPT = "A vaporwave computer"

response = client.images.generate(
    model="dall-e-3",
    prompt=PROMPT,
)

print(response.data[0].url)

与之前类似，这段代码向 API 发送了一个经过认证的请求，根据 PROMPT 中的文本生成一张图片。这段代码还做了一些优化，以便你在脚本中添加更多功能：

第 3 行初始化了 OpenAI 的一个实例，并将其存储为 client。这个实例已经包含了身份验证信息，因为你已经将环境变量命名为 OPENAI_API_KEY。如果你遵循了这个命名规则，它会自动从你的环境变量中获取 API 密钥值。当然，你也可以在创建实例时直接通过 api_key 参数传递 API 密钥。

第 5 行将文本提示定义为一个常量。这样做的好处是，如果你需要修改提示文本，只需要在脚本顶部找到这个常量并进行修改，非常方便。

第 7 行调用了 client 的 .images.generate() 方法。接下来的几行代码展示了你可以传递给这个方法的一些参数。

第 8 行指定使用更新的 DALL·E 3 模型，它不仅以不同的方式处理你的提示，还支持不同的参数设置、图片尺寸和质量，与默认的 DALL·E 2 模型有所区别。

第 9 行将 PROMPT 的值传递给名为 prompt 的参数。这样，你就为 DALL·E 提供了用于生成图片的文本。别忘了，你之前在命令行中调用 API 时也传递了文本提示。

最后，你还需要获取生成图片的 URL，以便在线查看。为此，你需要访问响应对象的 .url 属性，并在第 12 行将其值打印到终端。

当你执行这个脚本时，你将得到一个类似于之前的输出，但现在只会显示 URL，而不会显示完整的 JSON 响应。

(venv) $ python create.py
https://oaidalleapiprodscus.blob.core.windows.net/private/org-H8Satq5hV2d
⮑ scyutIStbN89m/user-O5O8mpE4TugEI9zqpxx8tGPt/img-hMAEPhuqVH73JMRQ6BOV4J
⮑ Hz.png?st=2024-08-13T10%3A56%3A54Z&se=2024-08-13T12%3A56%3A54Z&sp=r&sv
⮑ =2023-11-03&sr=b&rscd=inline&rsct=image/png&skoid=d505667d-d6c1-4a0a-b
⮑ ac7-5c84a87759f8&sktid=a48cca56-e6da-484e-a814-9c849652bcb3&skt=2024-0
⮑ 8-12T21%3A36%3A11Z&ske=2024-08-13T21%3A36%3A11Z&sks=b&skv=2023-11-03&s
⮑ ig=D7K6UF353uO3wJDOovbeyBsn2yaf384kEotVbjJ5SB4%3D

单击链接或将其粘贴到浏览器中以查看生成的图像。因为您现在使用的是 DALL·E 3 模型，所以输出可能看起来更引人注目：

这表明图像的质量和细节有了显著提高，看起来新模型对蒸汽波风格的理解更加深入。

虽然这很有可能，但在背后还有一些其他因素也在影响着图像细节的提升。当你请求使用 DALL·E 3 模型生成图像时，它会在生成图像前对你的提示进行重新编辑。

如果你想知道 DALL·E 在生成图像时实际使用的修改后的提示是什么，你可以通过 .revised_prompt 属性来查看。

# ...

print(response.data[0].revised_prompt)

在请求生成图像时，通常来说，提供更详细的描述会得到更理想的结果。在你的描述中加入更多的细节可以大幅提升效果！例如，下面是生成上面图像所用的详细描述：

想象一台带有蒸汽波风格的计算机；它像是一台80年代的老式机器，被霓虹粉和蓝色光晕所包围。它的设计充满了怀旧感，机身装饰着复古的字体。外壳上印有发光的数字棕榈树，让人联想到老式视频游戏中的场景。背景中可以看到希腊雕塑和蒸汽波设计中常见的故障日落图案，完美地结合了现代与复古的元素。

这比你最初提供的描述要详细得多！

注意：只有对 DALL·E 3 模型的请求才会进行提示重写，所以如果你使用的是 DALL·E 2，这个属性将不存在。

虽然 DALL·E 3 能够生成更令人印象深刻的图像，但它不允许你关闭提示重写功能，这意味着你对输出的控制会减少。此外，这个模型目前还没有完全开放所有功能。

因此，在本教程的后续部分，你将继续使用 DALL·E 2。首先，编辑脚本，在调用 .generate() 方法时添加一些额外的参数。同时，不妨也为你最初的描述增加一些细节：

from openai import OpenAI

client = OpenAI()

PROMPT = "An eco-friendly computer from the 90s in the style of vaporwave"

response = client.images.generate(
    model="dall-e-2",  # Default
    prompt=PROMPT,
    n=1,
    size="256x256",
)

print(response.data[0].url)

你已经为你的描述增加了一些细节，并切换回了默认的 DALL·E 2 模型。此外，你还添加了两个新参数来进一步定制生成的图像：

第 10 行将数字 1 指定给参数 n。这个参数允许你定义想要根据描述生成的新图像数量。n 的值应在 1 到 10 之间，默认为 1。

第 11 行设置了 size 参数的值。通过这个参数，你可以指定 DALL·E 生成图像的尺寸。这个参数需要是一个字符串，可用的尺寸会根据你使用的模型而有所不同。对于 DALL·E 2，可选的尺寸有 "256x256"、"512x512" 或 "1024x1024"。每个字符串都代表了你将收到的图像的像素尺寸，默认为最大的 "1024x1024"。

在你的脚本最后，你再次将生成图像的 URL 输出到终端。你已经移除了打印 .revised_prompt 的代码行，因为在 DALL·E 2 中这个值是不存在的。

当你执行脚本后，你会得到一个新的 URL 作为回应。点击这个链接或将其复制到你的浏览器中，就可以查看生成的图像了。和之前一样，你的图像会有所不同，但应该与你在 PROMPT 中使用的描述相似。由于你这次使用的是 DALL·E 2，所以图像的细节可能会比上一次少一些。

你可能会注意到，这次生成的图像比你之前创建的要小得多。这是因为你通过 size 参数指定了生成一张 256x256 像素的图片。较小的图像成本较低，所以你刚刚省下了一笔费用！作为一个节省成本的成功者，也许你还想保存其他东西——比如你的图像数据。

微信扫一扫分享文章