Pytorch学习笔记1.tensorboard

首先包含函数

from torch.utils.tensorboard import SummaryWriter

writer=SummaryWriter("logs")

此函数Creates a `SummaryWriter` that will write out events and summaries to the event file.

函数1：add_scalar（tag(str),scalar_value,global_step)

tag：标题

scalar_value：纵坐标

global_step：横坐标

测试1：

绘制y=x

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

检验方法：

在pycharm终端输入tensorboard --logdir="logs"命令，返回一个在线节点，进入即可看到。

---

函数2：add_image(tag, img_tensor(torch.Tensor, numpy.array), global_step,dataformats)

tag：标题

img_tensor：图片格式

global_step：测试步长

dataforms：图片数据格式（记得从控制台查看）

测试2：

打开一个图片

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

writer=SummaryWriter("logs")
img_path="dataset/hymenoptera_data/train/ants_image/69639610_95e0de17aa.jpg"//相对路径
img_PIL=Image.open(img_path)//打开
img_array=np.array(img_PIL)//格式转化为numpy格式

writer.add_image("test",img_array,2,dataformats='HWC')//输入参数
writer.close()

检验方法：

tensorboard --logdir="logs"

如果要改变图片，首先改变路径，再在add_image函数中修改标题（在tensorboard中新建一个）或修改步长（在同一图片中进行另一步长，可变化）