PyTorch 1.0 中文文档:torchvision.utils

2022-05-07 14:25:53 浏览数 (1)

译者:BXuan694

代码语言:javascript复制
torchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scale_each=False, pad_value=0)

把图片排列成网格形状。

参数:

  • tensorTensor list)– 四维批(batch)Tensor或列表。如果是Tensor,其形状应是(B x C x H x W);如果是列表,元素应为相同大小的图片。
  • nrowint, 可选)– 最终展示的图片网格中每行摆放的图片数量。网格的长宽应该是(B / nrow, nrow)。默认是8。
  • paddingint, 可选)– 扩展填充的像素宽度。默认是2。
  • normalizebool, 可选)– 如果设置为True,通过减去最小像素值然后除以最大像素值,把图片移到(0,1)的范围内。
  • rangetuple, 可选)– 元组(min, max),min和max用于对图片进行标准化处理。默认的,min和max由输入的张量计算得到。
  • scale_eachbool, 可选)– 如果设置为True,将批中的每张图片按照各自的最值分别缩放,否则使用当前批中所有图片的最值(min, max)进行统一缩放。
  • pad_valuefloat, 可选)– 扩展填充的像素值。

示例:

请看 这里

代码语言:javascript复制
torchvision.utils.save_image(tensor, filename, nrow=8, padding=2, normalize=False, range=None, scale_each=False, pad_value=0)

用于把指定的Tensor保存成图片文件。

参数:

阅读全文/改进本文

0 人点赞