pydalle：Python的DALL·E 2 API包装器

此库提供对DALL·E 2 API的基本程序访问。

此库的目的是为研究人员提供一种简单地将DALL·E 2的结果布局到jupyter笔记本或类似工具中的方法。

pydalle有两种主要的使用模式

• pydalle.Dalle：这是库的主要类。它提供了一个用户友好的界面来访问DALL·E 2 API。 更多信息。
• pydalle.imperative.api.labs：此模块提供了一组可以用于与DALL·E 2 API交互的底层函数。 更多信息。

安装

安装所有依赖项

pip install pydalle[all]     # Install all dependencies, recommended for most users

选择您需要的依赖项

pip install pydalle          # Just install the library with no optional dependencies
pip install pydalle[sync]    # Also installs requests (for synchronous networking)
pip install pydalle[async]   # Also installs aiohttp and aiofiles  (required for async networking / file handling)
pip install pydalle[images]  # Also installs Pillow and numpy (required for help with image processing)

技巧

• 通过注册 DALL·E 2等待列表 获取访问权限。

• 确保您的DALL·E 2使用遵守DALL·E 2的 内容政策 和 使用条款。

• 请注意，此库使您轻松花费金钱/DALL·E 2积分。

入门

安装pydalle后，您可以通过导入它并创建一个 Dalle 对象来开始使用它。您可以在 Dalle 类 上找到所有可用方法。

import os

from pydalle import Dalle

OPENAI_USERNAME = os.environ.get('OPENAI_USERNAME')
OPENAI_PASSWORD = os.environ.get('OPENAI_PASSWORD')


def main():
    client = Dalle(OPENAI_USERNAME, OPENAI_PASSWORD)
    print(f"Client created. {client.get_credit_summary().aggregate_credits} credits remaining...")
    tasks = client.get_tasks(limit=5)
    print(f"{len(tasks)} tasks found...")

    print("Attempting to download a generation of the first task and show off some built-in helpers...")
    if tasks and tasks[0].generations:
        example = tasks[0].generations[0].download()
        example.to_pil().show()  # Convert the image to a PIL image and show it
        example.to_pil_masked(x1=0.5, y1=0, x2=1, y2=1).show()  # Show a version with left side transparent (for edits)
        example.to_pil_padded(0.5).show()  # Show w/ 50% padding around the image, centered at (50%, 50%)
        example.to_pil_padded(0.4, cx=0.25, cy=0.25).show()  # Show w/ 40% padding, centered at (25%, 25%)

    print("Attempting to do a text2im task...")
    completed_text2im_task = client.text2im("A cute cat")
    for image in completed_text2im_task.download():
        image.to_pil().show()

    print("Attempting to create variations task on the first cat...")
    first_generation = completed_text2im_task.generations[0]
    completed_variation_task = first_generation.variations()
    first_variation = completed_variation_task.generations[0]
    first_image = first_variation.download()
    first_image.to_pil().show()

    print("Attempting to create inpainting task and showing the mask...")
    # Make the right-side of the image transparent
    mask = first_image.to_pil_masked(x1=0.5, y1=0, x2=1, y2=1)
    mask.show("inpainting mask")
    completed_inpainting_task = first_generation.inpainting("A cute cat, with a dark side", mask)
    for image in completed_inpainting_task.download():
        image.to_pil().show()


if __name__ == '__main__':
    main()

关于等效的异步代码示例，请参阅 examples/dev_client_async.py。

有关低级API示例以及在笔记本中使用这些示例的说明，请参阅 examples/low_level 目录。