Yolov5 使用Tensorboard可视化

在yolov5文件夹下的 ./models/yolo.py 中
找到注释过得tensorboard的代码（272行）

    # Tensorboard (not working https://github.com/ultralytics/yolov5/issues/2898)
    from torch.utils.tensorboard import SummaryWriter
    tb_writer = SummaryWriter('.')
    logger.info("Run 'tensorboard --logdir=models' to view tensorboard at http://localhost:6006/")
    tb_writer.add_graph(torch.jit.trace(model, img, strict=False), [])  # add model graph
    tb_writer.add_image('test', img[0], dataformats='CWH')  # add model to tensorboard

删除注释#注意缩进

不知道为啥这里是

logger.info(“Run ‘tensorboard --logdir=models’ to view tensorboard at http://localhost:6006/”)

但是运行的时候用的语句是

tensorboard --logdir=runs

先运行train.py 然后在命令行中输入上面那行，配置服务器与本地接口
使用Xshell7 见下图 隧道 里面，把6006端口配置一下

然后就可以在浏览器里输入：http://localhost:6006/