Diffusers 文档

实用程序

Hugging Face's logo
加入 Hugging Face 社区

并获得增强文档体验

开始使用

实用程序

用于处理 🤗 Diffusers 的实用程序和辅助函数。

numpy_to_pil

diffusers.utils.numpy_to_pil

< >

( images )

将 NumPy 图像或一批图像转换为 PIL 图像。

pt_to_pil

diffusers.utils.pt_to_pil

< >

( images )

将 Torch 图像转换为 PIL 图像。

load_image

diffusers.utils.load_image

< >

( image: Union convert_method: Optional = None ) PIL.Image.Image

参数

  • convert_method (Callable[[PIL.Image.Image], PIL.Image.Image], 可选) — 加载图像后应用于图像的转换方法。当设置为None时,图像将转换为“RGB”。

返回

PIL.Image.Image

一个 PIL 图像。

image加载到 PIL Image 中。

export_to_gif

diffusers.utils.export_to_gif

< >

( image: List output_gif_path: str = None fps: int = 10 )

export_to_video

diffusers.utils.export_to_video

< >

( video_frames: Union output_video_path: str = None fps: int = 10 )

make_image_grid

diffusers.utils.make_image_grid

< >

( images: List rows: int cols: int resize: int = None )

准备一个图像网格。用于可视化目的。

randn_tensor

diffusers.utils.torch_utils.randn_tensor

< >

( shape: Union generator: Union = None device: Optional = None dtype: Optional = None layout: Optional = None )

一个用于在所需的device上创建具有所需dtype的随机张量的辅助函数。当传递生成器列表时,可以单独为每个批次大小设置种子。如果传递了 CPU 生成器,则张量始终在 CPU 上创建。