简介
tprchvision是 P y T o r c h PyTorch PyTorch的一个图形库,其服务于 P y T o r c h PyTorch PyTorch的深度学习框架,主要用来构建计算机视觉模型,torchvision主要用于常见的一些图形变换,以下是 t o r c h v i s i o n torchvision torchvision的构成:
- 1.torchvision.datasets: 一些加载的数据函数及常用的数据接口.
- 2.torchvision.models: 包含常用的模型结构: (含预训练模型) 例如AlexNet、VGG、ResNet等;
- 3.torchvision.transforms: 常用的图片变换,裁剪旋转等.<