【Pytorch学习】TensorBoard

SummaryWriter类的使用

add_scalar()

from torch.utils.tensorboard import SummaryWriter

# 创建实例
writer = SummaryWriter("logs") # 存储的文件夹名为logs

# writer.add_image()

# y = x
for i in range(100):
    writer.add_scalar("y=x", i, i) # tag, y, x

writer.close()

打开TensorBoard，可以在Terminal终端处输入命令

tensorboard --logdir=logs --port=6007

其中，logs为存储文件夹名字，port为指定端口。因为一般会默认为打开的端口为6006，如果想要同时打开多个的话，就需要指定不同的端口打开。

add_image()

from torch.utils.tensorboard import SummaryWriter
from PIL import Image
import numpy as np

# 创建实例
writer = SummaryWriter("logs") # 存储的文件夹名为logs

image_path = "data/train/ants_image/0013035.jpg"
img_PIL = Image.open(image_path) # img_PIL为PIL类型
img_array = np.array(img_PIL) # img_array为numpy类型,并且格式shape为(H,W,C)

writer.add_image("test", img_array, 1, dataformats="HWC") # tag, image,

writer.close()