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

文章目录

    • 模型网络结构可视化之 netron
    • 工具总结
    • 网页版ONNX模型可视化测试
      • 操作如下
      • yolov5-s 可视化效果如下
      • yolov4-tiny.cfg.txt + yolov4-tiny.weights 可视化效果
    • Ubuntu 安装和服务启用
    • Mac M1 安装和服务启用
        • 点击下载
        • 拖拽安装即可
        • 打开程序即可使用
    • 精选专栏


模型网络结构可视化之 netron


GitHub 链接如下

https://github.com/lutzroeder/netron

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

https://netron.app/

支持客户端和网页操作

模型可视化工具Netron手把手教学_第1张图片

工具总结

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

网页版ONNX模型可视化测试


https://netron.app/

操作如下

模型可视化工具Netron手把手教学_第2张图片

yolov5-s 可视化效果如下

模型可视化工具Netron手把手教学_第3张图片

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

两个文件缺一不可

模型可视化工具Netron手把手教学_第4张图片

模型可视化工具Netron手把手教学_第5张图片

模型可视化工具Netron手把手教学_第6张图片


Ubuntu 安装和服务启用

模型可视化工具Netron手把手教学_第7张图片

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

模型可视化工具Netron手把手教学_第8张图片
模型可视化工具Netron手把手教学_第9张图片

Mac M1 安装和服务启用

点击下载

模型可视化工具Netron手把手教学_第10张图片
模型可视化工具Netron手把手教学_第11张图片

拖拽安装即可

模型可视化工具Netron手把手教学_第12张图片

打开程序即可使用

模型可视化工具Netron手把手教学_第13张图片

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

模型可视化工具Netron手把手教学_第14张图片


精选专栏


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

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

  • ❤️ 图像风格转换 —— 代码环境搭建 实战教程【关注即可阅】!
  • 图像修复-代码环境搭建-知识总结 实战教程 【据说还行】
  • 超分重建-代码环境搭建-知识总结 解秘如何让白月光更清晰【脱单神器】
  • YOLO专栏,只有实战,不讲道理 图像分类【建议收藏】!
  • 深度学习:环境搭建,一文读懂

  • 深度学习:趣学深度学习

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

  • CV 和 语音数据集:数据集整理

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

9-9

你可能感兴趣的:(工具使用-软件分享,人工智能,深度学习,模型可视化,可视化神器)