torchvision.utils
torchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scale_each=False, pad_value=0)
制作图像网格。
参数说明:
- tensor (Tensor or list) – 4D 小批量形状 Tensor 张量(B x C x H x W)或所有大小相同的图像列表。
- nrows (int, optional) – 网格中的行数。最终的网格大小(B / nrow,nrow)。默认值是 8。
- normalize (bool, optional) – 如果 TRUE,将图像移到范围(0, 1)中,减去最小值并除以最大像素值。
- range (tuple, optional) – 元组(min,max),其中 min 和 max 是数字,然后使用这些数字对图像进行标准化。默认情况下,min 和 max 是从张量计算的。
- scale_each (bool, optional) – 如果 TRUE,则在批处理图像中分别缩放每个图像,而不是在所有图像上(最小、最大)缩放图像。
- 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 张量保存到图像文件中。 参数说明:
- Tensor (张量或者列表) – 要保存的图像。如果小批量的 Tensor 张量,调用
make_grid
把 Tensor 张量存储为网格图像。 - kwargs – make_grid 的其他参数
译者署名
用户名 | 头像 | 职能 | 签名 |
---|---|---|---|
Song | 翻译 | 人生总要追求点什么 |