torchvision.utils

torchvision.utils.make_grid(tensor: Union[torch.Tensor, List[torch.Tensor]], nrow: int = 8, padding: int = 2, normalize: bool = False, range: Union[Tuple[int, int], NoneType] = None, scale_each: bool = False, pad_value: int = 0) → torch.Tensor

制作图像网格。

Parameters:

• 张量 (张量 或 列表) – 4D mini-batch 张量，形状为 (B x C x H x W) 或一个包含相同大小图像的列表。
• nrow (int, 可选) – 每行显示的图像数量。 最终网格大小是 (B / nrow, nrow)。默认值: 8。
• 填充 (int, 可选) – 填充量。默认值: 2。
• normalize (bool, 可选) – 如果为 True，则根据 range 指定的最小值和最大值将图像转换到 (0, 1) 范围。默认值: False。
• 范围 (元组, 可选) – 元组 (min, max) 其中 min 和 max 是数字， 然后这些数字用于归一化图像。默认情况下，min 和 max 是从张量计算得出的。
• scale_each (bool, 可选) – 如果为True，则对批次中的每张图像单独进行缩放，而不是对所有图像的（最小值，最大值）进行缩放。默认值：False。
• 填充值 (float, 可选) – 填充像素的值。默认值: 0。

示例

查看这个笔记本 在这里

torchvision.utils.save_image(tensor: Union[torch.Tensor, List[torch.Tensor]], fp: Union[str, pathlib.Path, BinaryIO], nrow: int = 8, padding: int = 2, normalize: bool = False, range: Union[Tuple[int, int], NoneType] = None, scale_each: bool = False, pad_value: int = 0, format: Union[str, NoneType] = None) → None

将给定的张量保存为图像文件。

Parameters:

• 张量 (张量 或 列表) – 要保存的图像。如果给定一个迷你批次张量， 则通过调用 make_grid 将张量作为图像网格保存。
• fp (字符串或文件对象) – 文件名或文件对象
• 格式 (可选) – 如果省略，则根据文件名扩展名确定要使用的格式。 如果使用文件对象而不是文件名，则应始终使用此参数。
• **kwargs – 其他参数请参见make_grid。