torchvision¶

The torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.

Package Reference

Datasets
- MNIST
- COCO
- LSUN
- ImageFolder
- Imagenet-12
- CIFAR
- STL10
- SVHN
- PhotoTour
Models
- Alexnet
- VGG
- ResNet
- SqueezeNet
- DensetNet
- Inception v3
Transforms
Utils