对CIFAR10中的图片利用tensorboard进行显示

最新推荐文章于 2023-09-27 15:43:19 发布

anzrs

最新推荐文章于 2023-09-27 15:43:19 发布

阅读量339

点赞数

文章标签： python

本文链接：https://blog.csdn.net/weixin_45193103/article/details/123809414

版权

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)

writer = SummaryWriter("p10")
#写入十张图片试试
for i in range(10):
    img , target = test_set[i]
    writer.add_image("test_set",img,i)
writer.close()