torchvision

最新推荐文章于 2024-10-12 20:31:18 发布

苏梓鑫鑫鑫

最新推荐文章于 2024-10-12 20:31:18 发布

阅读量126

点赞数 2

文章标签： pytorch 深度学习人工智能

本文链接：https://blog.csdn.net/auebud/article/details/140692062

版权

torchvision的使用介绍


import torchvision
from torch.utils.tensorboard import SummaryWriter

dataset_transform = torchvision.transforms.Compose([
    torchvision.transforms.ToTensor()
])

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)

#print(test_set[0])
#print(test_set.classes)
#
#img, target = test_set[0]
#print(img)
#print(targe)
#print(test_set.classes[target])
#img.show()

#print(test_set[0])

writer: SummaryWriter = SummaryWriter("P10")
for i in range(10):
    img, target = test_set[i]
    writer.add_image("test_set", img, i)

    writer.close()