当前位置: 首页 > 文档资料 > PyTorch 中文文档 >

torchvision.utils

优质
小牛编辑
132浏览
2023-12-01

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

猜测,用来做 雪碧图的sprite image)。

给定 4D mini-batch Tensor, 形状为 (B x C x H x W),或者一个a list of image,做成一个size(B / nrow, nrow)的雪碧图。

  • normalize=True ,会将图片的像素值归一化处理

  • 如果 range=(min, max), min和max是数字,那么minmax用来规范化image

  • scale_each=True ,每个图片独立规范化,而不是根据所有图片的像素最大最小值来规范化

Example usage is given in this notebook

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

将给定的Tensor保存成image文件。如果给定的是mini-batch tensor,那就用make-grid做成雪碧图,再保存。