pytorch模型结构可视化,可显示每层的尺寸

最近在学习一些检测方面的网络,使用的是pytorch。模型结构可视化是学习网络的有用的部分,pytorch没有原生支持这个功能,需要找一些其他方式,下面总结几种方法
(推荐用4)

1. torch .pt -> netron

netron是一个专门可视化模型的工具,支持很多格式,很自然的就是用它直接显示torch保存的模型。但是实际上…  

 

 

  显示成了上图,基本上没什么用。

2. 网上说的比较多的几种方式

a. tensorboardx 画出来的比较丑   b. tensorwatch 支持的torch版本低,容易出错   c. graphviz 实测yolov5异常。。

3. torch->caffe->netron

caffe支持的操作不够多,转换很容易碰到不支持的情况

4. torch->onnx->netron

pytorch对onnx支持的比较好,直接有导出接口。   torch.onnx.export…   但这样直接转出来的onnx进行显示没有每一层的输入大小,如下图,还是不方便看。  

 

 

  我们可以将导出的onnx进行以下操作:       import onnx   from onnx import shape_inference   model = 'best.onnx'   onnx.save(onnx.shape_inference.infer_shapes(onnx.load(model)), model)       再用netron进行显示即可得到,用这种结构图学习就好多了:  

 

 欢迎关注:老司机的视觉屋

   

给TA买糖
共{{data.count}}人
人已赞赏
经验教程

你知道 react-color 的实现原理吗

2021-1-10 20:12:47

经验教程

Java并发编程实战(4)- 死锁

2021-1-10 20:12:49

⚠️
免责声明:根据《计算机软件保护条例》第十七条规定“为了学习和研究软件内含的设计思想和原理,通过安装、显示、传输或者存储软件等方式使用软件的,可以不经软件著作权人许可,不向其支付报酬。”您需知晓本站所有内容资源均来源于网络,仅供用户交流学习与研究使用,版权归属原版权方所有,版权争议与本站无关,用户本人下载后不能用作商业或非法用途,需在24个小时之内从您的电脑中彻底删除上述内容,否则后果均由用户承担责任;如果您访问和下载此文件,表示您同意只将此文件用于参考、学习而非其他用途,否则一切后果请您自行承担,如果您喜欢该程序,请支持正版软件,购买注册,得到更好的正版服务。 本站为个人博客非盈利性站点,所有软件信息均来自网络,所有资源仅供学习参考研究目的,并不贩卖软件,不存在任何商业目的及用途,网站会员捐赠是您喜欢本站而产生的赞助支持行为,仅为维持服务器的开支与维护,全凭自愿无任何强求。本站部份代码及教程来源于互联网,仅供网友学习交流,若您喜欢本文可附上原文链接随意转载。
无意侵害您的权益,请发送邮件至 momeis6@qq.com 或点击右侧 私信:momeis 反馈,我们将尽快处理。
0 条回复 A文章作者 M管理员
    暂无讨论,说说你的看法吧
个人中心
今日签到
有新私信 私信列表
搜索