torchvision的数据集

最新推荐文章于 2024-09-05 20:34:00 发布

小六的画布

最新推荐文章于 2024-09-05 20:34:00 发布

阅读量250

点赞数

分类专栏： pytorch 文章标签：深度学习神经网络人工智能

本文链接：https://blog.csdn.net/qq_42202395/article/details/125765821

版权

pytorch 专栏收录该内容

28 篇文章 3 订阅

订阅专栏

torchvision的数据集
pytorch官网–>Docs–>torchvision

#p10_dataset_transforms
import torchvision
from torch.utils.tensorboard import SummaryWriter

dataset_transform = torchvision.tramsforms.Compose([
		torchvission.transforms.ToTenssor()
		])  # 转化totensor类型

train_set = torcchvision.datasets.CIFAR10(root="./dataset", train = Ture, transform = dataset_transform, download =Ture)
test_set = torcchvision.datasets.CIFAR10(root="./dataset", train = false, transform = dataset_transform,   download =Ture)
print(test_set[0])
writer = SummaryWriter("p10")
for i in range(10):
		img, target = test_set[i]
		writer.add_image("test_set", img, i)
writer.close()  #terminal中输入tensorboard --logdir="p10"
		
print(test_set[0])
print(test_set.classes)
img, target = test_set[0]
print(img)
print(target)
print(test_set.classes[target])
img.show()