pytorch自学笔记（5)

import torchvision
from tensorboardX import SummaryWriter#导入一些库

dataset_transform = torchvision.transforms.Compose([
    torchvision.transforms.ToTensor()
])
#此处还可以通过transform中的Compose进行一个数据的集合
train_set = torchvision.datasets.CIFAR10(root="dataset",train=True,transform=dataset_transform,download=True)
test_set = torchvision.datasets.CIFAR10(root="dataset",train=False,transform=dataset_transform,download=True)
#这里可以通过查看官方代码或者通过pycharm进行查看使用方法
# print(test_set[0])可以查看到其中包含一个calsses
# print(test_set.classes)
# img,target = test_set[0]#target相当于就是一个索引类型的东东
# print(img)#img就是一个图片
# print(target)
# print(test_set.classes[target])
# img.show()
# print(test_set[0])

writer = SummaryWriter("p10")
for i in range(10):
    img,target = test_set[i]
    writer.add_image("test_set",img,i)
#打印出来数据集中的图片

    今天同样学习了一些官方的数据集的一些操作等等，明天继续加油，新年快乐everybody！！！