Pytorch Note7 Dataset(数据集)
全部笔记的汇总贴:Pytorch Note 快乐星球
Dataset(数据集)
在处理任何机器学习问题之前都需要数据读取,并进行预处理。PyTorch提供了很多工具使得数据的读取和预处理变得很容易。
torchvision包 包含了目前流行的数据集,模型结构和常用的图片转换工具
torchvision.datasets中包含了以下数据集
- MNIST
- COCO(用于图像标注和目标检测)(Captioning and Detection)
- LSUN Classification
- ImageFolder
- Imagenet-12
- CIFAR10 and CIFAR100
- STL10
from torch.utils