enter image description here
为什么tensorboard显示不完整的图像
import torchvision
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter
test_data = torchvision.datasets.CIFAR10(root="../torvision/dataset_test", train=False,transform=torchvision.transforms.ToTensor())
test_loader = DataLoader(test_data, batch_size=64, shuffle=False, num_workers=0, drop_last=False)
writer = SummaryWriter("loader_log")
step = 0
for data in test_data:
img, target = data
writer.add_image("test_data", img, step)
step += 1
step_2 = 0
for data_2 in test_loader:
img_2, target_2 = data_2
writer.add_images("test_loader", img_2, step_2)
step_2 += 1
writer.close()
data_loader应该显示64个图像
1条答案
按热度按时间xkftehaa1#
test_data
,迭代数据集,并为每个图像单独添加一个唯一的标签。test_loader
,遍历数据加载器,并使用add_images将整批图像添加为网格。