Tensorboard的使用——04使用addimages将CIFAR10中的图片添加到Tensorboard中

Miles强

于 2022-05-15 14:35:44 发布

阅读量863

点赞数 3

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

版权

http://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz

1、设置transforms

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

2、下载数据集，使用DataLoader加载
batch_size指每次随机抓取64个图片

test_data = torchvision.datasets.CIFAR10(root="./test10_dataset", train=False, transform=dataset_transform)
test_loader = DataLoader(datase