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/
上一篇：
JS实现多线程数据分片下载
下一篇：
公司技术体系规划中的一些方向