Shortcuts

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[source]

制作一个图片网格。

Parameters:
  • tensor (Tensorlist) – 形状为 (B x C x H x W) 的4D小批量张量 或大小相同的图像列表。

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

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

  • normalize (bool, 可选) – 如果为True,则将图像移动到范围(0, 1), 通过value_range指定的最小值和最大值。默认值:False

  • value_range (tuple, optional) – 元组 (min, max),其中 min 和 max 是数字, 然后这些数字用于归一化图像。默认情况下,min 和 max 是从张量中计算得出的。

  • scale_each (bool, 可选) – 如果为 True,则分别缩放批次中的每个图像,而不是所有图像的(最小,最大)值。默认值:False

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

Returns:

包含图像网格的张量。

Return type:

网格 (张量)

使用 make_grid 的示例:

可视化工具

Visualization utilities