[pytorch入门] 4. torchvision中数据集的使用

介绍

文档

可以去看官方文档

可以在里面找到一些数据集的使用

CIFAR10

import torchvision
from torch.utils.tensorboard import SummaryWriter

dataset_transform = torchvision.transforms.Compose([
    torchvision.transforms.ToTensor(),
])

train_set = torchvision.datasets.CIFAR10(root = "./dataset",train=True, download=True)
test_set = torchvision.datasets.CIFAR10(root = "./dataset",train=False, download=True)

writer = SummaryWriter('logs')
for i in range(10):
    img, target = train_set[i]
    writer.add_image('train_set', img, i)

writer.close()

这就可以在项目目录下看到下载的压缩文件，若文件已经下载好，再次运行时不会重复下载
文件下载下来比较慢，可以复制链接用迅雷等工具下