1. 下载安装Netron
安装命令:
sudo dpkg -i netron_3.5.4_amd64.deb
2. 使用Netron进行网络可视化
首先在终端输入netron命令
接着打开基于pytorch的.pth模型,可以看到Netron工具对pytorch模型并不友好.
然后打开基于ONNX的.onnx模型,如下图所示
进而打开基于TensorFlow的.pb模型,如下图所示
最后打开基于Caffe的.caffemodel模型,如下图所示
安装命令:
sudo dpkg -i netron_3.5.4_amd64.deb
首先在终端输入netron命令
接着打开基于pytorch的.pth模型,可以看到Netron工具对pytorch模型并不友好.
然后打开基于ONNX的.onnx模型,如下图所示
进而打开基于TensorFlow的.pb模型,如下图所示
最后打开基于Caffe的.caffemodel模型,如下图所示