1. import torchvision
    2. from torch.utils.tensorboard import SummaryWriter
    4. trans_dataset = torchvision.transforms.Compose([
    5.     torchvision.transforms.ToTensor()
    6. ])
    8. train_set = torchvision.datasets.CIFAR10(root='./dataset',train=True,transform = trans_dataset,download=True)
    9. test_set = torchvision.datasets.CIFAR10(root='./dataset',train=False,transform = trans_dataset,download=True)
    11. print(test_set[0])
    12. print(test_set.classes)
    14. # img,target = test_set[0]
    15. # print(img)
    16. # print(type(img))
    17. # print(target)
    18. # print(test_set.classes[target])
    19. # img.show()
    21. writer = SummaryWriter('logs')
    23. for i in range(10):
    24.     img_tensor, target = test_set[i]
    25.     writer.add_image('torchvision_dataset',img_tensor,i)
    27. writer.close()
    29. #要学会看官方文档
    30. #CIFAR10   可以按住ctrl点击进入源代码查看需要输入的参数
    31. #也可以到pytorch官网查看  pytorch.org