三种实现模型可视化的方式(print, torchinfo, tensorboard)

最新推荐文章于 2024-04-01 18:01:28 发布

FrenchOldDriver

最新推荐文章于 2024-04-01 18:01:28 发布

阅读量998

点赞数 1

分类专栏：深度学习图像处理统计学/数据处理/机器学习文章标签： python 开发语言图像处理深度学习

本文链接：https://blog.csdn.net/OldDriver1995/article/details/130138878

版权

统计学/数据处理/机器学习同时被 3 个专栏收录

34 篇文章 60 订阅

订阅专栏

图像处理

15 篇文章 0 订阅

订阅专栏

深度学习

11 篇文章 1 订阅

订阅专栏

记录一下自己使用的三种模型可视化的方式，从简单到难

Print

最简单的是print，就不用多说了。

Torchinfo

from torchinfo import summary
import torch
model = (...)
summary(model, (1,3,128,128))

即可按照像文档路径一样的方式输出结构，并且有每一层的shape和参数量，比print的信息更多。
在这里插入图片描述
但相对而言还是比较简单。

Tensorboard

该方法是个人觉得最好用的并且很动态
需要提前下好tensorboard和tensorboardX

import torch
from torchvision.models import AlexNet
 
from tensorboardX import SummaryWriter
 
 
x=torch.rand(1,3,256,512)
model=AlexNet()
 
with SummaryWriter(comment='AlexNet') as w:
    w.add_graph(model, x)

运行该方法后会在同级路径下生成一个runs文件夹，其中就有每次的文件
在这里插入图片描述
再开一个终端，进入runs之后，输入指令tensorboard --logdir=XXX --port 8080 这里XXX就是runs下方某一个文件夹的名称。--port 指定端口。个人建议指定一个不常用的端口，因为有时常用端口会被其他程序占用，不一定能显示，遇到过类似的坑。运行了命令之后点开对应的链接，一般是localhost:port 或者127.0.0.1:port ，浏览器打开即可。
在这里插入图片描述
他这个图是动态的，点击你想看的模块可以放大，比如打开Alexnet，只要有模块就可以一直方法，很方便。