import torchvisionfrom 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！！！