torchvision.utils 文件

torchvision.utils.make_grid(张量：Union[torch.张量、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 （Tensor or list） – 形状为 （B x C x H x W） 的 4D 小批量张量 或大小相同的图像列表。
• nrow （int， optional） – 网格每行中显示的图像数。 最终网格大小为 。违约：。(B / nrow, nrow)8
• padding （int， optional） - 填充量。违约：。2
• normalize （bool，可选） – 如果为 True，则将图像移动到 （0， 1） 范围内， 按 指定的最小值和最大值。违约：。rangeFalse
• range （tuple， optional） – 元组 （min， max），其中 min 和 max 是数字， 然后，这些数字用于标准化图像。默认情况下，min 和 max 是从张量计算得出的。
• scale_each （bool， optional） – 如果 ，则缩放批次 images 而不是 （min， max） 覆盖所有图像。违约：。TrueFalse
• pad_value （float， optional） （ （float， optional） （浮点数，可选） – 填充像素的值。违约：。0

例

在此处查看此笔记本

torchvision.utils.save_image(张量：Union[torch.张量、List[torch.Tensor]]， fp： Union[str， pathlib.路径，BinaryIO]，nrow：int = 8，填充：int = 2，标准化：bool = False，范围：Union[Tuple[int， int]，NoneType] = None，scale_each：bool = False，pad_value：int = 0，格式：Union[str， NoneType] = None） → None

将给定的 Tensor 保存到图像文件中。

参数：

• tensor （Tensor or list） - 要保存的图像。如果给定一个 mini-batch 张量， 通过调用 将张量保存为图像网格。make_grid
• fp （string or file object） – 文件名或文件对象
• format （可选） – 如果省略，则要使用的格式由文件扩展名确定。 如果使用 file 对象而不是 filename，则应始终使用此参数。
• **kwargs – 其他参数记录在 中。make_grid