tensorborad的使用

上代码

import cv2 as cv
from torch.utils.tensorboard import SummaryWriter  #引入tensorboard模块

writer = SummaryWriter('log')	#执行此步后会在工作区中生成名为log的文件

image_path = "D:\\dataset\\data\\train\\bees_image\\16838648_415acd9e3f.jpg"

img = cv.imread(image_path)

writer.add_image('test', img, 1, dataformats="HWC")  #执行处会在log文件下生成对应的日志
writer.close()

代码解读

writer.add_image('test', img, 1, dataformats="HWC")

test为名字
1为步数step
add_image()函数默认shape为(C, H, W)，由于img的shape为(H, W, C)，
所有要用dataformats=“HWC”，指定为(H, W, C)，不然会报错

运行代码

运行后在ide的终端中输入

tensorboard --logdir=log

然后会出现如下链接

http://localhost:6006/

打开链接，即可看到运行结果

默认打开的是以上链接的端口

切换端口的方法

按CTEL+C退出
输入以下命令

tensorboard --logdir=log --port=6007

即可打开6007的端口
再想打开新端口在改6007为6008、6009等等即可