1.CIFAR-10数据集介绍
2.torchvision内含数据集的下载
from torchvision import datasets
from torchvision import transforms
from torch.utils.tensorboard import SummaryWriter
compose = transforms.Compose([
transforms.ToTensor()
])
train_dataset = datasets.CIFAR10(root="./datasets", train=True, transform=compose, download=True)
test_dataset = datasets.CIFAR10(root="./datasets", train=False, transform=compose, download=True)
writer = SummaryWriter("data")
for i in range(100):
img_tensor, label = train_dataset[i]
target = train_dataset.classes[label]
writer.add_image(tag="CIFAR-10", img_tensor=img_tensor, global_step=i)
writer.close()