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

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

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

第八集的主角是netron。

这有什么用?

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"

结束语

嘿嘿,很好用的,有帮助就点个赞和关注咯。感谢各位。
西瓜6的啦啦啦

猜你喜欢

转载自blog.csdn.net/qq_37924224/article/details/119207498