TensorBoard是一个可视化工具，它可以用来展示网络图、张量的指标变化、张量的分布情况等。

add_scalar()

  1. from torch.utils.tensorboard import SummaryWriter
  3. writer = SummaryWriter("log")
  5. # y = x
  6. for i in range(100):
  7.     writer.add_scalar("y=x", i, i)
  9. writer.close()

image.png

image.png

add_image()

  1. from torch.utils.tensorboard import SummaryWriter
  2. import numpy as np
  3. from PIL import Image
  5. writer = SummaryWriter("log")
  6. image_path = "数据集/dataset/train/ants_image/0013035.jpg"
  7. img = Image.open(image_path)
  8. img_array = np.array(img)
  10. writer.add_image("test", img_array, 1, dataformats="HWC")
  12. writer.close()

image.png

image.png

  1. from PIL import Image
  2. from torch.utils.tensorboard import SummaryWriter
  3. from torchvision import transforms
  5. image_path = "数据集/dataset/train/ants_image/0013035.jpg"
  6. img = Image.open(image_path)
  8. writer = SummaryWriter("log")
  10. # 1、transforms该如何使用
  11. tensor_trans = transforms.ToTensor()
  12. tensor_img = tensor_trans(img)
  14. writer.add_image("Tensor_img", tensor_img)
  16. writer.close()

image.png