就是来源于这里
import torchvision# 使用torchvision自带的数据集
train_set = torchvision.datasets.CIFAR10(root='./dataset', train=True, download=True)
test_set = torchvision.datasets.CIFAR10(root='./dataset', train=False, download=True)# 将下载好的测试集中的东西打印出来
print(test_set[0])
print(test_set.classes)img, target = test_set[0]
print(img)
print(target)
print(test_set.classes[target])img.show()
这里可以看到img,和target 的值是获取到了训练集中这个大类中对应的目标值
tensor数据类型的话就可以使用tensorboard进行输出使用
如何使用writer进行输出?
数据集和transform结合在一起
import torchvision# 使用transforms转换数据集
from torch.utils.tensorboard import SummaryWriterdataset_trans = torchvision.transforms.Compose([torchvision.transforms.ToTensor()
])# 使用torchvision自带的数据集
train_set = torchvision.datasets.CIFAR10(root='./dataset', train=True, transform=dataset_trans, download=True)
test_set = torchvision.datasets.CIFAR10(root='./dataset', train=False, transform=dataset_trans, download=True)# # 将下载好的测试集中的东西打印出来
# print(test_set[0])
# print(test_set.classes)
#
# img, target = test_set[0]
# print(img)
# print(target)
# print(test_set.classes[target])
# img.show()# 使用writer将获取到的图片进行输出
writer = SummaryWriter('p10')
for i in range(10):img, target = test_set[i]writer.add_image('test_set', img, i)writer.close()
D:\ANACONDA\envs\pytorch\python.exe C:/Users/Administrator/Desktop/Code/learn_pytorch/P14_dataset_transforms.py
Files already downloaded and verified
Files already downloaded and verified
Process finished with exit code 0
来吧展示
在官网中寻找自己想要的东西和答案