2018-04-07

PyTorch之torchvision

torchvision是PyTorch框架中的一个非常重要的包，由三个子包组成，如下：

• torchvision.datasets
• torchvision.models
• torchvision.transforms

1. torchvision.datasets

torchvision.datasets包含了以下数据集 ：

• MNIST
• COCO（用于图像标注和目标检测）(Captioning and Detection)
• LSUN Classification
• ImageFolder
• Imagenet-12
• CIFAR10 and CIFAR100
• STL10
  代码示例

import torchvision.datasets as dset
import torchvision.transforms as transforms