DL模型視覺化

2021-10-08 09:52:08 字數 1391 閱讀 1521

可能至少有5中方法:

model = alexnet2fc(

) x = torch.rand(8,

3,112,

112)

y = model(x)

# need tensorflow 2.2 or higher

# method 1

# from keras.utils import plot_model

# plot_model(model, to_file='model.png')

# method 2

# from ipython.display import svg

# from keras.utils.visualize_util import model_to_dot

# svg(model_to_dot(model).create(prog='dot', format='svg'))

from torchviz import make_dot

# method 3 for not mtl

# g = make_dot(y)

# g.render('net_arch', view=false)

# method 4 for mtl。模型的輸出是乙個列表,要麼拼接

for i in

range

(len

(y))

:if i ==

0: c = torch.cat(

(y[0

], y[1]

),1)

elif i >=

2and i <=

len(y)-2

: c = torch.cat(

(c, y[i+1]

),1)

g = make_dot(c)

g.render(

'net_arch'

, view=

false

)# 會自動儲存為乙個 espnet.pdf,第二個引數為true,則會自動開啟該pdf檔案,為false則不開啟

# method 5 for mtl,要麼生成乙個元祖,這種方法更推薦!!(模型輸出為長度40的列表)

g = make_dot(

tuple

((y[i]

for i in

range(40

))),

) g.render(

'net_arch'

, view=

false

)# 會自動儲存為乙個 espnet.pdf,第二個引數為true,則會自動開啟該pdf檔案,為false則不開啟

method 1/2:

method 5來自github神人:chesharma @

caffe模型視覺化

通過前面的學習,我們已經能夠正常訓練各種資料了。設定好solver.prototxt後,我們可以把訓練好的模型儲存起來,如lenet iter 10000.caffemodel。訓練多少次就自動儲存一下,這個是通過snapshot進行設定的,儲存檔案的路徑及檔名字首是由snapshot prefix...

TensorBoard模型視覺化

tensorboard是乙個基於瀏覽器的互動式工具,可以讓我們看到學習過程,並探索我們訓練好的模型。要執行tensorboard,首先到命令終端 開始 anaconda anaconda prompt,輸入activate tensorflow 然後,告訴tensorboard記錄的相關摘要 ten...

視覺視覺化模型

盒模型 規定單個盒子的規則 視覺視覺化模型 布局規則 頁面中的多個盒子的排列規則 視覺視覺化模型,大體上將頁面上的盒子分為三種排列規則 常規流 浮動 定位 常規流 文件流 普通文件流 常規文件流 所有元素,預設情況下 都屬於常規流布局 總體規則 塊盒獨佔一行 行盒水平一次排列 包含塊 contain...