一、CIFAR-10数据集
是由 Hinton 的学生 Alex Krizhevsky 和 Ilya Sutskever 整理的一个用于识别普适物体的小型数据集。CIFAR-10数据集包含60000幅32x32的彩色图像,分为10个类,每类6000幅图像。训练图像50000张,测试图像10000张。
数据集下载地址:
http://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
二、代码
1、设置transforms
dataset_transform = torchvision.transforms.Compose([
torchvision.transforms.ToTensor()
])
2、下载数据集,划分为训练集和测试集
如果数据集已经提前下载好download填False
train_set = torchvision.datasets.CIFAR10(root="./test10_dataset", train=True,transform=dataset_transform