torchvision中数据集的使用

最新推荐文章于 2023-03-08 16:58:50 发布

luck060606

最新推荐文章于 2023-03-08 16:58:50 发布

阅读量474

点赞数

本文链接：https://blog.csdn.net/luck060606/article/details/122808005

版权

深度学习机器学习 pytorch

import torchvision
import ssl
from torch.utils.tensorboard import SummaryWriter
ssl._create_default_https_context = ssl._create_unverified_context
dataset_transform=torchvision.transforms.ToTensor()
train_set=torchvision.datasets.CIFAR10(root='E:\\postgraduate\\torchexer',train=True,
                                       transform=dataset_transform,download=False)
test_set=torchvision.datasets.CIFAR10(root='E:\\postgraduate\\torchexer',train=False,
                                      transform=dataset_transform,download=False)
#img,target=train_set[0]
#img.show()

#print(train_set[0])

writer=SummaryWriter('logs')
for i in range(10):
    img,target=train_set[i]
    writer.add_image('1',img,i)
writer.close()