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()