torchvision
Posted #self-discipline#
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了torchvision相关的知识，希望对你有一定的参考价值。
tochvision是pytorch中用于图像处理的库。
torchvision由以下部分组成：
torchvision.datasets：加载数据集
torchvision.models：提供训练模型
torchvision.transforms：图形变换方法（裁剪，转向等）
data_transform = transforms.Compose(
[transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])
transforms.compose将图像操作链接起来
torchvision.utils：其它方法
以上是关于torchvision的主要内容，如果未能解决你的问题，请参考以下文章
torchvision.transforms.Compose()详解Pytorch入门手册