当前位置:   article > 正文

可视化工具Netron介绍

netron

      Netron是一种用于神经网络、深度学习和机器学习模型的可视化工具,它可以为模型的架构生成具有描述性的可视化(descriptive visualization)。源码在:https://github.com/lutzroeder/netron ,主要由JavaScript语言实现,License为MIT,最新发布版本为5.9.6。

      Netron是一个跨平台工具,可以在Linux、Windows和Mac上运行,并且支持多种框架和格式Netron支持ONNX、TensorFlow Lite、Caffe、Keras、Darknet、PaddlePaddle、ncnn、MNN、Core ML、RKNN、MXNet、MindSpore Lite、TNN、Barracuda、Tengine、CNTK、TensorFlow.js、Caffe2 和 UFF。它还实验性支持PyTorch、TensorFlow、TorchScript、OpenVINO、Torch、Vitis AI、kmodel、Arm NN、BigDL、Chainer、Deeplearning4j、MediaPipe、ML.NET 和 scikit-learn。

      安装Netron:

      (1).windows:从https://github.com/lutzroeder/netron/releases/tag/v5.9.6 下载Netron-Setup-5.9.6.exe,双击安装即可,安装完后会在桌面生成Netron的图标,双击此图标即可打开,界面如下图所示:

      (2)Linux:从https://github.com/lutzroeder/netron/releases/tag/v5.9.6 下载Netron-5.9.6.AppImage,给此文件添加执行权限:chmod u+x Netron-5.9.6.AppImage ,然后双击此文件即可打开。

      示例:打开一个onnx模型,如Lenet-5.onnx(来自于:https://blog.csdn.net/fengbingchun/article/details/126072998 ),点击Open Model...按钮,结果如下图所示:点击左侧的每一个方框(类似于layer类型名),都会在右侧显示该框属性等信息

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/655989
推荐阅读
  

闽ICP备14008679号