《用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"
结束语
嘿嘿,很好用的,有帮助就点个赞和关注咯。感谢各位。