一、什么是TorchVision
torchvision是pytorch的一个图形库,用来处理图像,主要用来构建计算机视觉模型。
从下面的官网截图可以看到torchvision有很多模块,下面以dataset模块进行举例。
torchvision中datasets包:用来进行数据加载,主要有以下几个模块
CelebA
CIFAR
Cityscapes
COCO
Captions
Detection
DatasetFolder
EMNIST
FakeData
Fashion-MNIST
Flickr
HMDB51
ImageFolder
ImageNet
Kinetics-400
KMNIST
LSUN
MNIST
Omniglot
PhotoTour
Places365
QMNIST
SBD
SBU
STL10
SVHN
UCF101
USPS
VOC