tensorboard的快捷使用方法

tensorboard使用方法

建议不要使用docker来进行使用，因为直接使用的是文件，所以可以直接使用外部的环境。使用pip install tensorboard 进行安装。然后使用tensorboard --logdir log路径 --port 6007然后修改xshell里面的属性里面的隧道。

代码里面的操作：首先 from tensorboardX import SummaryWriter
在log处

log = Logger()
log.write(boneName)
writer = SummaryWriter(log_dir=’…/Logs/’)

需要定义一个步数：
if global_step % 5 == 0:
writer.add_scalar(‘Loss/loss’, loss_var[‘loss’], global_step)
writer.add_scalar(‘Loss/xy_loss’, loss_var[‘xy_loss’], global_step)
writer.add_scalar(‘Loss/wh_loss’, loss_var[‘wh_loss’], global_step)
writer.add_scalar(‘Loss/giou_loss’, loss_var[‘giou_loss’], global_step)
writer.add_scalar(‘Loss/cls_loss’, loss_var[‘cls_loss’], global_step)
writer.add_scalar(‘Loss/conf_loss’, loss_var[‘conf_loss’], global_step)
writer.add_scalar(‘Loss/ts_loss’, loss_var[‘ts_loss’], global_step)
writer.add_scalar(‘lr’, optimizer.param_groups[0][“lr”], global_step)