import torchvision
dataset_transform = torchvision.transforms.Compose([torchvision.transforms.ToTensor()])
train_set = torchvision.datasets.CIFAR10(root="./dataset_CIFAR10", train=True, transform=dataset_transform, download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset_CIFAR10", train=False, transform=dataset_transform, download=True)
print(test_set[0])
# print(test_set.classes)
#
# img, label = test_set[0]
# print(img)
# print(label)
Pytorch学习(4)torchvision里数据集的使用
最新推荐文章于 2024-05-17 10:41:35 发布