pytorch -- tensorboard使用

斑驳的岁月2024-02-26 14:38

1. 导入

from torch.utils.tensorboard import SummaryWriter
2. ctrl键点击查看如何使用
3. writer = SummaryWriter("dir") 对应事件文件输出到文件夹dir中
4. def add_image(
self,
tag,
img_tensor,
global_step=None, 步骤
walltime=None, dataformats="CHW"
):
5. writer.add_scalar("y=x",i,i) 添加标量数据到summary中，tag:图表标题
def add_scalar(
self,
tag, 图表的标题
scalar_value, y轴数值
global_step=None, x轴数值
walltime=None,
new_style=False,
double_precision=False,
):
6. 打开事件文件终端使用： tensorboard --logdir=logs

使用：

python 复制代码

from torch.utils.tensorboard import SummaryWriter
import cv2
writer = SummaryWriter("logs")


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

img_path = "dataset/train/bees/16838648_415acd9e3f.jpg"
img_cv = cv2.imread(img_path)
print(img_cv.shape)
writer.add_image("ants",img_cv,1, dataformats='HWC')

writer.close()