安装torchvision
Anaconda:
conda install torchvision -c soumith
pip:
pip install torchvision
torchvision 主要包括:
1. vision.datasets 常用视觉数据集的数据加载器,例如MNIST、COCO、Imagenet-12、CIFAR10 以及CIFAR100等
2. vision.models:流行的模型体系结构,例如AlexNet,VGG和ResNet以及预先训练的模型
3. vision.transforms:常见的图像变换,如随机裁剪,旋转等
4. vision.utils