模型可视化工具Netron手把手教学

文章目录

模型网络结构可视化之 netron

GitHub 链接如下

https://github.com/lutzroeder/netron

网页版工具直接拖拽使用地址

https://netron.app/

支持客户端和网页操作

工具总结

• 这个工具主要是帮助我们可视化网络模型层结构，可以观察到每个层的name、输入输出shape，有时候和模型本身代码对照来看，就会更容易看明白每一层的含义
• 有时候，在推理时，通过观察这个可视化结构，比较容易看出来网络是否有冗余的层（推理时不需要）

网页版ONNX模型可视化测试

https://netron.app/

操作如下

yolov5-s 可视化效果如下

yolov4-tiny.cfg.txt + yolov4-tiny.weights 可视化效果

两个文件缺一不可

Ubuntu 安装和服务启用

浏览器 打开 http://localhost:8080 ，即可开始本地使用这个模型可视化工具啦

Mac M1 安装和服务启用

点击下载

拖拽安装即可

打开程序即可使用

打开一个 生成器，看起来还是很丝滑的

📙 精选专栏

• 🍊 深度学习模型训练推理——基础环境搭建推荐博文查阅顺序【基础安装—认真帮大家整理了】——【1024专刊】

计算机视觉领域 八大专栏、不少干货、有兴趣可了解一下

• ❤️ 图像风格转换 —— 代码环境搭建 实战教程【关注即可阅】！

• 💜 图像修复-代码环境搭建-知识总结 实战教程 【据说还行】

• 💙 超分重建-代码环境搭建-知识总结 解秘如何让白月光更清晰【脱单神器】

• 💛 YOLO专栏，只有实战，不讲道理 图像分类【建议收藏】！

• 🍊 深度学习：环境搭建，一文读懂

• 🍊 深度学习：趣学深度学习

• 🍊 落地部署应用：模型部署之转换-加速-封装

• 🍊 CV 和 语音数据集：数据集整理

• 🍊 点赞 👍 收藏 ⭐留言 📝 都是博主坚持写作、更新高质量博文的最大动力！