Buckets:

hf-doc-build/doc / diffusers /v0.43.0 /zh /api /image_processor.md
mishig's picture
|
download
raw
24.5 kB
# VAE 图像处理器
[VaeImageProcessor](/docs/diffusers/v0.43.0/zh/api/image_processor#diffusers.VaeImageProcessor) 为 [StableDiffusionPipeline](/docs/diffusers/v0.43.0/zh/api/pipelines/stable_diffusion/text2img#diffusers.StableDiffusionPipeline) 提供了统一的 API,用于在 VAE 编码前准备图像输入,以及在解码后对输出进行后处理。这包括调整大小、归一化,以及在 PIL Image、PyTorch 和 NumPy 数组之间进行转换等操作。
所有使用 [VaeImageProcessor](/docs/diffusers/v0.43.0/zh/api/image_processor#diffusers.VaeImageProcessor) 的 pipeline 均接受 PIL Image、PyTorch tensor 或 NumPy 数组作为图像输入,并根据用户提供的 `output_type` 参数返回相应的输出。您可以直接将编码后的图像 latents 传入 pipeline,并通过 `output_type` 参数(例如 `output_type="latent"`)将 latents 作为特定输出从 pipeline 返回。这使得您可以将一个 pipeline 生成的 latents 直接作为输入传递给另一个 pipeline,而无需离开 latent 空间。此外,通过在不同 pipeline 之间直接传递 PyTorch tensor,使用多个 pipeline 协同工作也变得更加简单。
## VaeImageProcessor[[diffusers.VaeImageProcessor]]
#### diffusers.VaeImageProcessor[[diffusers.VaeImageProcessor]]
```python
diffusers.VaeImageProcessor(do_resize: bool = True, vae_scale_factor: int = 8, vae_latent_channels: int = 4, resample: str = 'lanczos', reducing_gap: int | None = None, do_normalize: bool = True, do_binarize: bool = False, do_convert_rgb: bool = False, do_convert_grayscale: bool = False)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L83)
**参数:**
do_resize (`bool`,*可选*,默认为 `True`):是否将图像的(高度,宽度)维度缩小为 `vae_scale_factor` 的倍数。可以接受来自 [image_processor.VaeImageProcessor.preprocess()](/docs/diffusers/v0.43.0/zh/api/image_processor#diffusers.VaeImageProcessor.preprocess) 方法的 `height` 和 `width` 参数。
vae_scale_factor (`int`,*可选*,默认为 `8`):VAE 缩放因子。如果 `do_resize` 为 `True`,图像将自动调整为该因子的整数倍。
resample (`str`,*可选*,默认为 `lanczos`):调整图像大小时使用的重采样滤波器。
do_normalize (`bool`,*可选*,默认为 `True`):是否将图像归一化到 [-1,1]。
do_binarize (`bool`,*可选*,默认为 `False`):是否将图像二值化为 0/1。
do_convert_rgb (`bool`,*可选*,默认为 `False`):是否将图像转换为 RGB 格式。
do_convert_grayscale (`bool`,*可选*,默认为 `False`):是否将图像转换为灰度格式。
VAE 的图像处理器。
#### apply_overlay[[diffusers.VaeImageProcessor.apply_overlay]]
```python
apply_overlay(mask: Image, init_image: Image, image: Image, crop_coords: tuple[int, int, int, int] | None = None)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L789)
**参数:**
mask (`PIL.Image.Image`):用于突出显示需要叠加区域的遮罩图像。
init_image (`PIL.Image.Image`):应用叠加层的原始图像。
image (`PIL.Image.Image`):要叠加到原始图像上的图像。
crop_coords (`tuple[int, int, int, int]`,*可选*):用于裁剪图像的坐标。如果提供,图像将按此进行裁剪。
**返回:** `PIL.Image.Image`
应用叠加层后的最终图像。
在原始图像上叠加显示掩码和修复后的图像。
#### binarize[[diffusers.VaeImageProcessor.binarize]]
```python
binarize(image: Image)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L524)
**参数:**
image (`PIL.Image.Image`):图像输入,应为 PIL 图像。
**返回:** `PIL.Image.Image`
二值化图像。小于 0.5 的值被设置为 0,大于 0.5 的值被设置为 1。
创建一个掩码。
#### 模糊[[diffusers.VaeImageProcessor.blur]]
```python
blur(image: Image, blur_factor: int = 4)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L271)
**参数:**
image (`PIL.Image.Image`):要转换为灰度图的 PIL 图像。
**返回:** `PIL.Image.Image`
经过灰度转换的 PIL 图像。
对图像应用高斯模糊。
#### convert_to_grayscale[[diffusers.VaeImageProcessor.convert_to_grayscale]]
```python
convert_to_grayscale(image: Image)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L254)
**参数:**
image (`PIL.Image.Image`):要转换的输入图像。
**返回:** `PIL.Image.Image`
转换为灰度图像后的图像。
将给定的 PIL 图像转换为灰度图像。
#### convert_to_rgb[[diffusers.VaeImageProcessor.convert_to_rgb]]
```python
convert_to_rgb(image: Image)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L237)
**参数:**
image (`PIL.Image.Image`):要转换为 RGB 的 PIL 图像。
**返回:** `PIL.Image.Image`
经过 RGB 转换的 PIL 图像。
将 PIL 图像转换为 RGB 格式。
#### 反规范化[[diffusers.VaeImageProcessor.denormalize]]
```python
denormalize(images: typing.Union[numpy.ndarray, torch.Tensor])
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L222)
**参数:**
images (`np.ndarray` 或 `torch.Tensor`):要反归一化的图像数组。
**返回:** `np.ndarray` 或 `torch.Tensor`
反归一化的图像数组。
将图像数组反归一化到 [0,1] 范围。
#### get_crop_region[[diffusers.VaeImageProcessor.get_crop_region]]
```python
get_crop_region(mask_image: Image, width: int, height: int, pad = 0)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L288)
**参数:**
mask_image (PIL.Image.Image) : 掩码图像。
width (int):要处理的图像的宽度。
height (int):要处理的图像的高度。
pad (int, 可选) : 要添加到裁剪区域的填充。默认值为 0。
**返回:** `tuple`
(x1, y1, x2, y2) 表示一个矩形区域,该区域包含图像中所有被掩码的区域,并且与原始宽高比相匹配。
查找包含图像中所有掩码区域的矩形区域,并将该区域扩展以匹配原始图像的宽高比;例如,如果用户在 128x32 的区域中绘制了掩码,而处理尺寸为 512x512,则该区域将被扩展为 128x128。
#### get_default_height_width[[diffusers.VaeImageProcessor.get_default_height_width]]
```python
get_default_height_width(image: typing.Union[PIL.Image.Image, numpy.ndarray, torch.Tensor], height: int | None = None, width: int | None = None)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L561)
**参数:**
image (`PIL.Image.Image | np.ndarray | torch.Tensor`):图像输入,可以是 PIL 图像、NumPy 数组或 PyTorch 张量。如果是 NumPy 数组,其形状应为 `[batch, height, width]` 或 `[batch, height, width, channels]`。如果是 PyTorch 张量,其形状应为 `[batch, channels, height, width]`。
height (`int | None`,*可选*,默认为 `None`):预处理图像的高度。如果为 `None`,则使用 `image` 输入的高度。
width (`int | None`,*可选*,默认为 `None`):预处理图像的宽度。如果为 `None`,则使用 `image` 输入的宽度。
**返回:** `tuple[int, int]`
一个元组,包含高度和宽度,两者均调整为 `vae_scale_factor` 的最近整数倍。
返回图像的高度和宽度,向下取整为 `vae_scale_factor` 的下一个整数倍。
#### normalize[[diffusers.VaeImageProcessor.normalize]]
```python
normalize(images: typing.Union[numpy.ndarray, torch.Tensor])
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L207)
**参数:**
images (`np.ndarray` 或 `torch.Tensor`):要归一化的图像数组。
**返回:** `np.ndarray` 或 `torch.Tensor`
归一化后的图像数组。
将图像数组归一化到 [-1,1]。
#### numpy_to_pil[[diffusers.VaeImageProcessor.numpy_to_pil]]
```python
numpy_to_pil(images: ndarray)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L128)
**参数:**
images (`np.ndarray`) : 要转换为 PIL 格式的图像数组。
**返回:** `list[PIL.Image.Image]`
PIL 图像列表。
将 numpy 图像或一批图像转换为 PIL 图像。
#### numpy_to_pt[[diffusers.VaeImageProcessor.numpy_to_pt]]
```python
numpy_to_pt(images: ndarray)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L172)
**参数:**
images (`np.ndarray`) : 要转换为 PyTorch 格式的 NumPy 图像数组。
**返回:** `torch.Tensor`
图像的 PyTorch 张量表示。
将 NumPy 图像转换为 PyTorch 张量。
#### pil_to_numpy[[diffusers.VaeImageProcessor.pil_to_numpy]]
```python
pil_to_numpy(images: list[PIL.Image.Image] | PIL.Image.Image)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L152)
**参数:**
images (`PIL.Image.Image` 或 `list[PIL.Image.Image]`):要转换为 NumPy 格式的 PIL 图像或图像列表。
**返回:** `np.ndarray`
图像的 NumPy 数组表示。
将 PIL 图像或 PIL 图像列表转换为 NumPy 数组。
#### postprocess[[diffusers.VaeImageProcessor.postprocess]]
```python
postprocess(image: Tensor, output_type: str = 'pil', do_denormalize: list[bool] | None = None)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L739)
**参数:**
image (`torch.Tensor`) : 图像输入,应为形状为 `B x C x H x W` 的 PyTorch 张量。
output_type (`str`,*可选*,默认为 `pil`):图像的`pil`、`np`、`pt`、`latent`输出类型之一。
do_denormalize (`list[bool]`,*可选*,默认为 `None`):是否将图像反归一化到 [0,1]。如果为 `None`,将使用 `VaeImageProcessor` 配置中 `do_normalize` 的值。
**返回:** `PIL.Image.Image`、`np.ndarray` 或 `torch.Tensor`
后处理后的图像。
将图像输出从张量后处理为`output_type`。
#### preprocess[[diffusers.VaeImageProcessor.preprocess]]
```python
preprocess(image: typing.Union[PIL.Image.Image, numpy.ndarray, torch.Tensor, list[PIL.Image.Image], list[numpy.ndarray], list[torch.Tensor]], height: int | None = None, width: int | None = None, resize_mode: str = 'default', crops_coords: tuple[int, int, int, int] | None = None)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L608)
**参数:**
image (`PipelineImageInput`):图像输入,支持的格式包括 PIL 图像、NumPy 数组、PyTorch 张量;也接受上述支持格式的列表。
height (`int`, *可选*) : 预处理图像的高度。如果为 `None`,将使用 `get_default_height_width()` 获取默认高度。
width (`int`, *可选*) : 预处理后的宽度。如果为 `None`,将使用 get_default_height_width()` 获取默认宽度。
resize_mode (`str`, *可选*,默认为 `default`) : 调整大小模式,可以是 `default` 或 `fill` 之一。如果为 `default`,则将图像调整大小以适应指定的宽度和高度,且可能不保持原始宽高比。如果为 `fill`,则将图像调整大小以适应指定的宽度和高度,保持宽高比,然后将图像居中于该尺寸内,并用图像数据填充空白区域。如果为 `crop`,则将图像调整大小以适应指定的宽度和高度,保持宽高比,然后将图像居中于该尺寸内,并裁剪掉多余部分。请注意,resize_mode `fill` 和 `crop` 仅支持 PIL 图像输入。
crops_coords (`list[tuple[int, int, int, int]]`,*可选*,默认为 `None`):批次中每张图片的裁剪坐标。如果为 `None`,则不对图片进行裁剪。
**返回:** `torch.Tensor`
预处理后的图像。
预处理图像输入。
#### pt_to_numpy[[diffusers.VaeImageProcessor.pt_to_numpy]]
```python
pt_to_numpy(images: Tensor)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L191)
**参数:**
images (`torch.Tensor`) : 要转换为 NumPy 格式的 PyTorch 张量。
**返回:** `np.ndarray`
图像的 NumPy 数组表示。
将 PyTorch 张量转换为 NumPy 图像。
#### resize[[diffusers.VaeImageProcessor.resize]]
```python
resize(image: typing.Union[PIL.Image.Image, numpy.ndarray, torch.Tensor], height: int, width: int, resize_mode: str = 'default')
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L463)
**参数:**
image (`PIL.Image.Image`、`np.ndarray` 或 `torch.Tensor`):图像输入,可以是 PIL 图像、numpy 数组或 pytorch 张量。
height (`int`) : 要调整到的高度。
width (`int`) : 调整后的宽度。
resize_mode (`str`,*可选*,默认为 `default`):要使用的缩放模式,可以是 `default` 或 `fill` 之一。如果为 `default`,则将图像缩放以适应指定的宽度和高度,且可能不保持原始宽高比。如果为 `fill`,则将图像缩放以适应指定的宽度和高度,保持宽高比,然后将图像居中于该尺寸内,并用图像数据填充空白区域。如果为 `crop`,则将图像缩放以适应指定的宽度和高度,保持宽高比,然后将图像居中于该尺寸内,并裁剪掉多余部分。请注意,resize_mode `fill` 和 `crop` 仅支持 PIL 图像输入。
**返回:** `PIL.Image.Image`、`np.ndarray` 或 `torch.Tensor`
调整大小后的图像。
调整图像大小。
## InpaintProcessor[[diffusers.InpaintProcessor]]
[InpaintProcessor](/docs/diffusers/v0.43.0/zh/api/image_processor#diffusers.InpaintProcessor) 接受 `mask` 和 `image` 输入并一起处理它们。可选地,它可以接受 padding_mask_crop 并应用 mask 叠加。
#### diffusers.InpaintProcessor[[diffusers.InpaintProcessor]]
```python
diffusers.InpaintProcessor(do_resize: bool = True, vae_scale_factor: int = 8, vae_latent_channels: int = 4, resample: str = 'lanczos', reducing_gap: int | None = None, do_normalize: bool = True, do_binarize: bool = False, do_convert_grayscale: bool = False, mask_do_normalize: bool = False, mask_do_binarize: bool = True, mask_do_convert_grayscale: bool = True)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L837)
用于图像和掩码修复的图像处理器。
#### postprocess[[diffusers.InpaintProcessor.postprocess]]
```python
postprocess(image: Tensor, output_type: str = 'pil', original_image: PIL.Image.Image | None = None, original_mask: PIL.Image.Image | None = None, crops_coords: tuple[int, int, int, int] | None = None)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L944)
对图像进行后处理,可选地应用遮罩叠加
#### preprocess[[diffusers.InpaintProcessor.preprocess]]
```python
preprocess(image: Image, mask: PIL.Image.Image | None = None, height: int | None = None, width: int | None = None, padding_mask_crop: int | None = None)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L882)
预处理图像和掩码。
## VaeImageProcessorLDM3D[[diffusers.VaeImageProcessorLDM3D]]
[VaeImageProcessorLDM3D](/docs/diffusers/v0.43.0/zh/api/image_processor#diffusers.VaeImageProcessorLDM3D) 接受 RGB 和深度输入,并返回 RGB 和深度输出。
#### diffusers.VaeImageProcessorLDM3D[[diffusers.VaeImageProcessorLDM3D]]
```python
diffusers.VaeImageProcessorLDM3D(do_resize: bool = True, vae_scale_factor: int = 8, resample: str = 'lanczos', do_normalize: bool = True)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L974)
**参数:**
do_resize (`bool`,*可选*,默认为 `True`):是否将图像的 (height, width) 维度缩小为 `vae_scale_factor` 的整数倍。
vae_scale_factor (`int`,*可选*,默认为 `8`):VAE 缩放因子。如果 `do_resize` 为 `True`,图像将自动调整为该因子的整数倍。
resample (`str`,*可选*,默认为 `lanczos`):调整图像大小时使用的重采样滤波器。
do_normalize (`bool`,*可选*,默认为 `True`):是否将图像归一化到 [-1,1]。
VAE LDM3D 的图像处理器。
#### depth_pil_to_numpy[[diffusers.VaeImageProcessorLDM3D.depth_pil_to_numpy]]
```python
depth_pil_to_numpy(images: list[PIL.Image.Image] | PIL.Image.Image)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L1025)
**参数:**
images (`list[PIL.Image.Image, PIL.Image.Image]`) : 要转换的输入图像或图像列表。
**返回:** `np.ndarray`
转换后的图像组成的 NumPy 数组。
将 PIL 图像或 PIL 图像列表转换为 NumPy 数组。
#### numpy_to_depth[[diffusers.VaeImageProcessorLDM3D.numpy_to_depth]]
```python
numpy_to_depth(images: ndarray)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L1083)
**参数:**
images (`np.ndarray`):输入的 NumPy 深度图像数组,可以是单张图像或一个批次。
**返回:** `list[PIL.Image.Image]`
从输入的 NumPy 深度图像转换而来的 PIL 图像列表。
将 NumPy 深度图像或一批图像转换为 PIL 图像列表。
#### numpy_to_pil[[diffusers.VaeImageProcessorLDM3D.numpy_to_pil]]
```python
numpy_to_pil(images: ndarray)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L1001)
**参数:**
images (`np.ndarray`) : 输入的 NumPy 图像数组,可以是单张图像或一个批次。
**返回:** `list[PIL.Image.Image]`
从输入 NumPy 数组转换而来的 PIL 图像列表。
将 NumPy 图像或一批图像转换为 PIL 图像列表。
#### preprocess[[diffusers.VaeImageProcessorLDM3D.preprocess]]
```python
preprocess(rgb: typing.Union[torch.Tensor, PIL.Image.Image, numpy.ndarray], depth: typing.Union[torch.Tensor, PIL.Image.Image, numpy.ndarray], height: int | None = None, width: int | None = None, target_res: int | None = None)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L1161)
**参数:**
rgb (`torch.Tensor | PIL.Image.Image | np.ndarray`) : RGB 输入图像,可以是单张图像或一个批次。
depth (`torch.Tensor | PIL.Image.Image | np.ndarray`):深度输入图像,可以是单张图像或一个批次。
height (`int | None`,*可选*,默认为 `None`):处理后图像的目标高度。如果为 `None`,则默认为输入图像的高度。
width (`int | None`,*可选*,默认为 `None`):处理后图像的期望宽度。如果为 `None`,则默认为输入图像的宽度。
target_res (`int | None`,*可选*,默认为 `None`):用于调整图像大小的目标分辨率。如果指定,将覆盖 height 和 width。
**返回:** `tuple[torch.Tensor, torch.Tensor]`
一个元组,包含处理后的 RGB 图像和深度图像,均为 PyTorch 张量。
预处理图像输入。接受的格式为 PIL 图像、NumPy 数组或 PyTorch 张量。
#### rgblike_to_depthmap[[diffusers.VaeImageProcessorLDM3D.rgblike_to_depthmap]]
```python
rgblike_to_depthmap(image: typing.Union[numpy.ndarray, torch.Tensor])
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L1045)
将类似 RGB 的深度图像转换为深度图。
## PixArtImageProcessor[[diffusers.PixArtImageProcessor]]
#### diffusers.PixArtImageProcessor[[diffusers.PixArtImageProcessor]]
```python
diffusers.PixArtImageProcessor(do_resize: bool = True, vae_scale_factor: int = 8, resample: str = 'lanczos', do_normalize: bool = True, do_binarize: bool = False, do_convert_grayscale: bool = False)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L1381)
**参数:**
do_resize (`bool`,*可选*,默认为 `True`):是否将图像的(高度,宽度)维度缩小为 `vae_scale_factor` 的倍数。可以接受来自 [image_processor.VaeImageProcessor.preprocess()](/docs/diffusers/v0.43.0/zh/api/image_processor#diffusers.VaeImageProcessor.preprocess) 方法的 `height` 和 `width` 参数。
vae_scale_factor (`int`,*可选*,默认为 `8`):VAE 缩放因子。如果 `do_resize` 为 `True`,图像将自动调整为该因子的整数倍。
resample (`str`,*可选*,默认为 `lanczos`):调整图像大小时使用的重采样滤波器。
do_normalize (`bool`,*可选*,默认为 `True`):是否将图像归一化到 [-1,1]。
do_binarize (`bool`,*可选*,默认为 `False`):是否将图像二值化为 0/1。
do_convert_rgb (`bool`,*可选*,默认为 `False`):是否将图像转换为 RGB 格式。
do_convert_grayscale (`bool`,*可选*,默认为 `False`):是否将图像转换为灰度格式。
用于 PixArt 图像缩放和裁剪的图像处理器。
#### classify_height_width_bin[[diffusers.PixArtImageProcessor.classify_height_width_bin]]
```python
classify_height_width_bin(height: int, width: int, ratios: dict)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L1422)
**参数:**
height (`int`) : 图像的高度。
width (`int`) : 图像的宽度。
ratios (`dict`) : 一个字典,其中键为宽高比,值为 (height, width) 元组。
**返回:** `tuple[int, int]`
最接近的分箱高度和宽度。
根据宽高比返回分箱后的高度和宽度。
#### resize_and_crop_tensor[[diffusers.PixArtImageProcessor.resize_and_crop_tensor]]
```python
resize_and_crop_tensor(samples: Tensor, new_width: int, new_height: int)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L1440)
**参数:**
samples (`torch.Tensor`) : 形状为 (N, C, H, W) 的张量,其中 N 为批次大小,C 为通道数,H 为高度,W 为宽度。
new_width (`int`) : 期望的输出图像宽度。
new_height (`int`) : 期望的输出图像高度。
**返回:** `torch.Tensor`
包含调整大小并裁剪后的图像的张量。
将图像张量调整大小并裁剪到指定尺寸。
## IPAdapterMaskProcessor[[diffusers.IPAdapterMaskProcessor]]
#### diffusers.IPAdapterMaskProcessor[[diffusers.IPAdapterMaskProcessor]]
```python
diffusers.IPAdapterMaskProcessor(do_resize: bool = True, vae_scale_factor: int = 8, resample: str = 'lanczos', do_normalize: bool = False, do_binarize: bool = True, do_convert_grayscale: bool = True)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L1277)
**参数:**
do_resize (`bool`,*可选*,默认为 `True`):是否将图像的 (height, width) 维度缩小为 `vae_scale_factor` 的整数倍。
vae_scale_factor (`int`,*可选*,默认为 `8`):VAE 缩放因子。如果 `do_resize` 为 `True`,图像将自动调整为该因子的整数倍。
resample (`str`,*可选*,默认为 `lanczos`):调整图像大小时使用的重采样滤波器。
do_normalize (`bool`,*可选*,默认为 `False`):是否将图像归一化到 [-1,1]。
do_binarize (`bool`,*可选*,默认为 `True`):是否将图像二值化为 0/1。
do_convert_grayscale (`bool`,*可选*,默认为 `True`):是否将图像转换为灰度格式。
用于 IP Adapter 图像蒙版的图像处理器。
#### 下采样[[diffusers.IPAdapterMaskProcessor.downsample]]
```python
downsample(mask: Tensor, batch_size: int, num_queries: int, value_embed_dim: int)
```
[源代码](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/image_processor.py#L1318)
**参数:**
mask (`torch.Tensor`) : 使用 `IPAdapterMaskProcessor.preprocess()` 生成的输入掩码张量。
batch_size (`int`) : 批次大小。
num_queries (`int`) : 查询的数量。
value_embed_dim (`int`):值嵌入的维度。
**返回:** `torch.Tensor`
下采样后的掩码张量。
将提供的掩码张量下采样,以匹配缩放点积注意力所需的预期维度。如果掩码的宽高比与输出图像的宽高比不匹配,则会发出警告。

Xet Storage Details

Size:
24.5 kB
·
Xet hash:
28e5ac6802344efa403bdbfea09aec3f0d4513a46fd8bc0d278a715d4a683138

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.