0
点赞
收藏
分享

微信扫一扫

torchvision中数据集的使用

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()
举报

相关推荐

0 条评论