可視化神經網絡模型的工具Netron


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了。
在這里插入圖片描述


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM