torchvision.utils

torchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scale_each=False, pad_value=0)

制作图像网格。

参数说明:

1. tensor (Tensor or list) – 4D小批量形状Tensor张量(B x C x H x W）或所有大小相同的图像列表。
2. nrows (int, optional) – 网格中的行数。最终的网格大小(B / nrow，nrow）。默认值是8。
3. normalize (bool, optional) – 如果TRUE，将图像移到范围(0, 1）中，减去最小值并除以最大像素值。
4. range (tuple, optional) – 元组(min，max），其中min和max是数字，然后使用这些数字对图像进行标准化。默认情况下，min和max是从张量计算的。
5. scale_each (bool, optional) – 如果TRUE，则在批处理图像中分别缩放每个图像，而不是在所有图像上(最小、最大）缩放图像。
6. pad_value (float, optional) – 填充像素的值。

查看下面的例子：

torchvision.utils.save_image(tensor, filename, nrow=8, padding=2, normalize=False, range=None, scale_each=False, pad_value=0)

将给定的Tensor张量保存到图像文件中。 参数说明:

1. Tensor (张量或者列表) – 要保存的图像。如果小批量的Tensor张量，调用make_grid把Tensor张量存储为网格图像。
2. kwargs – make_grid的其他参数

译者署名

用户名	头像	职能	签名
Song		翻译	人生总要追求点什么