pytorch 张力板显示不完整图像

wwodge7n  于 2023-10-20  发布在  其他
关注(0)|答案(1)|浏览(152)

enter image description here
为什么tensorboard显示不完整的图像

  1. import torchvision
  2. from torch.utils.data import DataLoader
  3. from torch.utils.tensorboard import SummaryWriter
  4. test_data = torchvision.datasets.CIFAR10(root="../torvision/dataset_test", train=False,transform=torchvision.transforms.ToTensor())
  5. test_loader = DataLoader(test_data, batch_size=64, shuffle=False, num_workers=0, drop_last=False)
  6. writer = SummaryWriter("loader_log")
  7. step = 0
  8. for data in test_data:
  9. img, target = data
  10. writer.add_image("test_data", img, step)
  11. step += 1
  12. step_2 = 0
  13. for data_2 in test_loader:
  14. img_2, target_2 = data_2
  15. writer.add_images("test_loader", img_2, step_2)
  16. step_2 += 1
  17. writer.close()

data_loader应该显示64个图像

xkftehaa

xkftehaa1#

test_data,迭代数据集,并为每个图像单独添加一个唯一的标签。
test_loader,遍历数据加载器,并使用add_images将整批图像添加为网格。

  1. import torchvision
  2. from torch.utils.data import DataLoader
  3. from torch.utils.tensorboard import SummaryWriter
  4. test_data = torchvision.datasets.CIFAR10(root="../torvision/dataset_test", train=False, transform=torchvision.transforms.ToTensor())
  5. test_loader = DataLoader(test_data, batch_size=64, shuffle=False, num_workers=0, drop_last=False)
  6. writer = SummaryWriter("loader_log")
  7. # Add images from the test_data one by one
  8. for i, (img, target) in enumerate(test_data):
  9. writer.add_image(f"test_data/{i}", img, i)
  10. # Add images from the test_loader as batches
  11. for i, (batch_img, batch_target) in enumerate(test_loader):
  12. writer.add_images("test_loader", batch_img, i)
  13. writer.close()
展开查看全部

相关问题