PyTorch③---torchvision中数据集的使用

torchvision中数据集的使用

import torchvision
from torch.utils.tensorboard import SummaryWriter


#转成tensor数据类型
dataset_transform=torchvision.transforms.Compose([
    torchvision.transforms.ToTensor()
])
#root="./dataset_CIFAR10"是下载路径
train_set=torchvision.datasets.CIFAR10(root="./dataset_CIFAR10",train=True,transform=dataset_transform,download=True)
test_set=torchvision.datasets.CIFAR10(root="./dataset_CIFAR10",train=False,transform=dataset_transform,download=True)
print(test_set[0])
print(test_set.classes)
print("---------------------------------------------------")
img,target=test_set[0]
print(img)
print(target)
print(test_set.classes[target])
#img.show()
print("---------------------------------------------------")
print(test_set[0])

writer=SummaryWriter("P10")
for i in range(10):
    img,target=test_set[i]
    writer.add_image("test_set",img,i)
writer.close()