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模型,如下图所示