Pytorch学习（三）torchvision中数据集的使用

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数据集：一般是用来进行目标检测和语义分割的。