39 Star 235 Fork 71

OpenDocCN / pytorch-doc-zh

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
torchvision-utils.md 963 Bytes
一键复制 编辑 原始数据 按行查看 历史
片刻小哥哥 提交于 2020-03-04 14:15 . 更新 1.4 翻译版本

torchvision.utils

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做成雪碧图,再保存。

1
https://gitee.com/OpenDocCN/pytorch-doc-zh.git
git@gitee.com:OpenDocCN/pytorch-doc-zh.git
OpenDocCN
pytorch-doc-zh
pytorch-doc-zh
master

搜索帮助