Pytorch
torchvision介绍
torchvision是pytorch的一个图形库,它服务于PyTorch深度学习框架的,主要用来构建计算机视觉模型。torchvision.transforms主要是用于常见的一些图形变换。以下是torchvision的构成:
torchvision.datasets: 一些加载数据的函数及常用的数据集接口
torchvison.models :包含常用的模型结构(含预训练模型),例如AlexNet , VGG, ResNet
torchvison.transforms: 常用的图片变换,例如裁剪,旋转等
torchvison.utils :其他一些有用的方法
1、transforms函数解析:torchvision.transforms 是pytorch中的图像预处理包,一般用Compose把多个图片变换步骤整合到一起。(实际是个列表,而这个列表里面的元素就是你想要执行的transform操作。Compose()类会将transforms列表里面的transform操作进行遍历。)如下所示
-
import torch import torchvision.transforms as transforms torch_transforms =