1. 下载安装Netron

下载链接: https://github.com/lutzroeder/netron/releases/tag/v3.5.4
安装命令:
sudo dpkg -i netron_3.5.4_amd64.deb
2. 使用Netron进行网络可视化
首先在终端输入netron命令


接着打开基于pytorch的.pth模型,可以看到Netron工具对pytorch模型并不友好.

然后打开基于ONNX的.onnx模型,如下图所示

进而打开基于TensorFlow的.pb模型,如下图所示

最后打开基于Caffe的.caffemodel模型,如下图所示

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/184704.html原文链接:https://javaforall.net
