netron的github地址:https://github.com/lutzroeder/Netron
- 使用netron可視化工具可以清晰地看見神經網絡的各層輸入名和輸出名以及具體的網絡結構。相比tensorboard它更加輕量化,而且支持各種框架:
- 各平臺安裝方式:
- 測試netron。新建一個test.py文件,並與xxx.onnx在同一個目錄下:
import netron
modelPath = "xxx.onnx"
netron.start(modelPath)
在終端執行:
python test.py
這時候點擊終端輸出的網址就可以在瀏覽器查看網絡結構了:
http://localhost:8080