torchvision

  • torchvision.utils.save_image save a given tensor to an image file
def save_image(tensor, filename, nrow=8, padding=2, 
       normalize=False, range=None, scale_each=False, pad_value=0)

if tensor is on cuda, it will be moved to ``CPU``` automatically.

  • torchvision.utils.make_grid generate sprite image (multi-rows, multi-cols)
def make_grid(tensor, nrow=8, padding=2, normalize=False, 
      range=None, scale_each=False, pad_value=0)

tensor: 4D mini-batch tensor [B, C, H, W] or a list of images with the same size
padding: padding between images.
normalize: if True, shift the image to the range (0, 1) by a specific range specified by range
range: tuple (min, max), if this None, minandmax are calculated from the tensor.
scale_each: scale each image separately not use range calculated from all images.

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容