2018-04-07
2018-04-07 本文已影响0人
shaoxiangjun
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
