可视化工具netron的简单使用

Posted 2023-04-18

参考技术A 在预训练的模型文件同一目录下新建一个python文件netrontest.py

在同一目录下打开命令行窗口执行python文件

这是会自动打开浏览器到地址 http://localhost:8080
显示网络图。此时即可查看输入输出或其中层的信息。

《用python的话，一定要试着使用的函数》第八集。（netron：超简单的神经网络模型可视化工具）

《用python的话，一定要试着使用的函数》专栏的第八集。这集是关于python的一个超简单的神经网络模型可视化工具。
这个专栏会确保短小精悍，学得快，看了不后悔。

用python的话，一定要试着使用的函数

第八集的主角是netron。

目录

• 用python的话，一定要试着使用的函数
• • 这有什么用?
  • 怎么装?
  • 怎么用？
  • 注意
  • 结束语

这有什么用?

Netron 是神经网络、深度学习和机器学习模型的查看器。
Netron 支持 ONNX、TensorFlow Lite、Keras、Caffe、Darknet、ncnn、MNN、PaddlePaddle、Core ML、MXNet、RKNN、MindSpore Lite、TNN、Barracuda、Tengine、TensorFlow.js、Caffe2 和 UFF。
Netron 对 PyTorch、TensorFlow、TorchScript、OpenVINO、Torch、Vitis AI、Arm NN、BigDL、Chainer、CNTK、Deeplearning4j、MediaPipe、ML.NET 和 scikit-learn 提供实验性支持。

怎么装?

pip install psutil

怎么用？

新建一个py文件，命名为"my_netron.py"(千万不要傻傻得命名为"netron.py")

接着三段代码

import netron
modelPath = "模型地址"
netron.start(modelPath)

运行后就会自动打开 http://localhost:8080/

注意

千万不要傻傻得命名为"netron.py"

结束语

嘿嘿，很好用的，有帮助就点个赞和关注咯。感谢各位。