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