【菜鸟学习笔记】Pytorch Tensorboard可视化工具的安装与简单使用步骤

1. 进入自己的pytorch环境，并安装任意版本的tensorflow(因为Tensorboard本来是给tensorflow使用的）

(base) C:\Users\Lenovo>conda activate mypytorch
(mypytorch) C:\Users\Lenovo>pip install tensorflow

2. 安装TensorboardX

(mypytorch) C:\Users\Lenovo>pip install TensorboardX

安装完成，下面以PFNet（CVPR2021）为例，使用TensorboardX对网络模型进行可视化。

3. 在训练代码中，实例化网络模型之后，加入代码:

# TensorboardX可视化网络结构
    # ①实例化SummaryWriter对象
    writer1 = SummaryWriter()  # 保存日志文件路径，默认使用runs/日期
    # ②将模型写入tensorboard
    init_img = torch.zeros((1, 3, 224, 224), device=device)
    writer1.add_graph(net, init_img) # net为实例化model    

4. 执行代码，会自动生成文件。右击该文件夹，复制其绝对路径
   
5. 在自己的pytorch环境下，执行tensorboard --logdir=“刚才复制的绝对路径”（注意：一定要加双引号！一定要是对应时间文件夹的绝对路径！），执行完毕后，如下图
   
6. 打开浏览器，输入上图url，即可查看网络结构。双击查看详细结构。
   

学习地址：
https://www.bilibili.com/video/BV1ha4y1Y7zv?spm_id_from=333.337.search-card.all.click&vd_source=00d61d2f6625d4f012f01cbf917378ca
https://www.bilibili.com/video/BV1Qf4y1C7kz?spm_id_from=333.337.search-card.all.click&vd_source=00d61d2f6625d4f012f01cbf917378ca