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 小批量 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 的示例

可视化实用工具

可视化实用工具