import torchvision
train_set = torchvision.datasets.CIFAR10(root = "", train = True, download = True)
test_set = torchvision.datasets.CIFAR10(root = "", train = False, download = True)
# 查看Dataset中的第一个数据集
print(test_set[0])
链接放入迅雷下载更快。
CAFAR-10中包括60000张32×32的彩色图片。一共十类,每类有6000张图片,其中5000张是训练集,1000张是测试集。
COCO数据集:一般是用来进行目标检测和语义分割的。