make_grid

torchvision.utils.make_grid(tensor: Union[Tensor, List[Tensor]], nrow: int = 8, padding: int = 2, normalize: bool = False, value_range: Optional[Tuple[int, int]] = None, scale_each: bool = False, pad_value: float = 0.0) → Tensor

生成图像网格。

参数:

• tensor (Tensor 或 list) – 形状为 (B x C x H x W) 的 4D mini-batch Tensor，或尺寸相同的图像列表。

• nrow (int, 可选) – 网格中每行显示的图像数量。最终网格尺寸为 (B / nrow, nrow)。默认值: 8。

• padding (int, 可选) – 填充量。默认值: 2。

• normalize (bool, 可选) – 如果为 True，则根据 value_range 指定的最小值和最大值，将图像值归一化到 (0, 1) 范围。默认值: False。

• value_range (tuple, 可选) – (min, max) 元组，其中 min 和 max 是数字，这些数字用于对图像进行归一化。默认情况下，min 和 max 是从 tensor 中计算得出的。

• scale_each (bool, 可选) – 如果为 True，则分别缩放图像批次中的每张图像，而不是使用所有图像的 (min, max) 进行缩放。默认值: False。

• pad_value (float, 可选) – 填充像素的值。默认值: 0。

返回:

包含图像网格的 tensor。

返回类型:

grid (Tensor)

使用 make_grid 的示例

可视化工具

可视化工具