使用教程位置:pytorch官网(https://pytorch.org/)→docs→torchvision
Dataset和transform的联合使用
下载数据集
# 下载数据集
train_set = torchvision.datasets.CIFAR10(root="./dataset_cifar10", train=True, transform=dataset_transforms, download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset_cifar10", train=False, transform=dataset_transforms, download=True)
print(test_set[0])
print(test_set.classes)
img, target = test_set[0]
print(img)
print(target)
print(test_set.classes[target])
与transform联合使用:
ataset_transforms = torchvision.transforms.Compose([
torchvision.transforms.ToTensor()
])
# 下载数据集,也可用迅雷下载,更快,下载完复制到当前目录下
train_set = torchvision.datasets.CIFAR10(root="./dataset_cifar10", train=True, transform=dataset_transforms, download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset_cifar10", train=False, transform=dataset_transforms, download=True)
print(test_set[0])
print(test_set.classes)
img, target = test_set[0]
print(img)
print(target)
print(test_set.classes[target])
writer = SummaryWriter("logs")
for i in range(10):
img,taget = test_set[i]
writer.add_image("test_set", img, i)
writer.close()