pytorch学习（三）：torchvision中的数据集的使用

LightningJie

已于 2024-05-16 11:21:42 修改

阅读量122

点赞数 2

分类专栏： pytorch 文章标签： pytorch 学习人工智能

于 2024-05-16 11:19:55 首次发布

本文链接：https://blog.csdn.net/weixin_52307528/article/details/138912240

版权

pytorch 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

以CIFAR数据集为例：

import torchvision
from torch.utils.tensorboard import SummaryWriter
dataset_transform=torchvision.transforms.Compose([
    torchvision.transforms.ToTensor(),

])
train_set=torchvision.datasets.CIFAR10('./dataset',train=True,transform=dataset_transform,download=True)
test_set=torchvision.datasets.CIFAR10('./dataset',train=False,transform=dataset_transform,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()
# print(test_set[0])
writer=SummaryWriter('p10')
for i in range(10):
    img,target=test_set[i]
    writer.add_image('test_img',img,i)
writer.close()