TensorBoard入门

一、显示图片

from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
# 通常使用writer作为类名：
writer = SummaryWriter("logs")
# 图片路径
image_path = "F:\PyCharm\learn_pytorch\\test2.jpg"
# 根据上述路径打开图片
img_PIL = Image.open(image_path)
# 将图片保存为numpy类型
img_array = np.array(img_PIL)
# 展示图片
writer.add_image("test", img_array, 1, dataformats='HWC')
# 关闭writer()
writer.close()

结果如下

二、显示图表

from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
# 通常使用writer作为类名：
writer = SummaryWriter("logs")
# y = 2x
for i in range(100):
    # 参数分别为：图像"标题" y x
    writer.add_scalar("y=2x", 2*i, i)
# 关闭writer()
writer.close()

结果如下