torchvision中数据集的使用

一、数据集下载

pytorch自带数据集，可直接通过方法并设置参数进行下载，以CIFAR10为例：

官方文档

import torchvision
from torch.utils.tensorboard import SummaryWriter
#将数据集PIL类型转换为Tensor类型
tensor_dataset = torchvision.transforms.Compose([torchvision.transforms.ToTensor()])

#数据集下载
train_set = torchvision.datasets.CIFAR10(root="./dataset",train=True,transform=tensor_dataset,download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset",train=False,transform=tensor_dataset,download=True)

writer = SummaryWriter("p10")
#将测试数据集中的十个图片在tensorboard进行展示
      for i in range(10):
            img, target = test_set[i]
            writer.add_image("test_set",img,i)
writer.close()

。。。。。。