1、介紹
最近一直在研究神經網絡,實現論文,搭建模型,有時候不清楚每層設置的參數,十分頭疼。偶然發現了一個可視化模型的工具Netron,在windows,mac,linux上都可以直接安裝,也支持web上瀏覽,十分方便,這里簡單記錄下。Windows安裝版的如下圖所示。
2、使用方法
web瀏覽器:https://lutzroeder.github.io/netron/
github地址:https://github.com/lutzroeder/netron
安裝包:https://github.com/lutzroeder/netron/releases
除了以上使用方法,當然還可以配合python使用。
在anaconda下命令行安裝pip install netron。
安裝后,首先導入netron,再加載模型,netron.start(path)
import netron
import os
model_path = os.path.join(os.getcwd(),"model","train_iter_40000.caffemodel")
netron.start(model_path)
運行啟動之后,在瀏覽器上輸入http://localhost:8080/,就OK了。