torchvision

import torchvision
train_set = torchvision.datasets.CIFAR10(root="./dataset",train=True,download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset",train=False,download=True)
# train = True 训练集 False 测试集
# root 指定了一个相对路径，直接下载到目录下兵命名为dataset
# download = True 下载
print(test_set[0])                      #查看测试集的第一个数据集，第一个图片，但target不是0
print(test_set.classes)
img,target = test_set[0]
print(img)               
print(target)
print(test_set.classes[target])
img.show()                              #图片为PIL格式打开方法

torchvision和transform联动

import torchvision
from torch.utils.tensorboard import SummaryWriter

dataset_transform = torchvision.transforms.Compose([torchvision.transforms.ToTensor()])
# 将图片转换为totensor格式
train_set = torchvision.datasets.CIFAR10(root="./dataset",train=True,transform=dataset_transform,download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset",train=False,transform=dataset_transform,download=True)

writer = SummaryWriter("p10")
for i in range(10):
    img,targe = test_set[i]
    writer.add_image("test_set",img,i)

writer.close()

之后在teiminal终端打开