网站与网页区别/cms网站模板
哔哩大学的PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】
的P13讲讲述了CIFAR10数据集的下载使用。
代码注释如下:
import torchvision# CIFAR10这个数据集,可以去官网上查看,包括32*32的像素图片,很多类别和图片,有训练和测试图片
train_set = torchvision.datasets.CIFAR10(root="./dataset", train=True, download=True)
# 按ctrl+P看函数需要什么,
# root即数据集需要存放的位置,在P10的地方创建一个dataset文件夹
# train默认是ture
# transform暂时不设置
# download=True意思为下载
test_set = torchvision.datasets.CIFAR10(root="./dataset", train=False, download=True)
# 运行即下载一个压缩文件并自动解压,下载慢可以把下载链接放到迅雷下载,迅雷有镜像加速print(test_set[0]) # 查看一下测试集第一个的输出
# 输出为(<PIL.Image.Image image mode=RGB size=32x32 at 0x2BDBF836AC8>, 3)
# 3代表target,tese_set中有classes,把图片属性转换为数字
print(test_set.classes)
# 此为把属性打印出来['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']img, target = test_set[0]
# img=<PIL.Image.Image image mode=RGB size=32x32 at 0x2BDBF836AC8>,target=3
# img, target分别获得了输出的两个属性
print(img)
print(target)
# 输出验证一下img, target
print(test_set.classes[target])
img.show()
结果如图: