神經網絡模型可視化工具netron

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章