【深度學習入門到精通系列】模型結構可視化神器Netron(連.pth都可以~!)

文章目錄

1 Netron

目前的Netron支持主流各種框架的模型結構可視化工作,我直接給出gayhub鏈接:
https://github.com/lutzroeder/Netron
支持windows,Linux,mac系統
在windows系統,下載一個.exe文件就很穩了,如下:
在這裏插入圖片描述

2 說明

安裝以後,就是隻需雙擊打開,添加模型文件的位置就可以了。
Netron 支持的框架和對應文件如下:
在這裏插入圖片描述
只需按照上述表格將對應的文件路徑加到netron中,就可以看到漂亮的網絡結構圖了,截取部分顯示:
在這裏插入圖片描述

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