2015-07-20 73 views
5

在CONV-網模型,我知道如何以可視化的過濾器,我們可以做itorch.image(型號:得到(1)。重量)可視化圖像,中間層在火炬(LUA)

但如何我可以在卷積後有效地顯示輸出圖像嗎?尤其是深層神經網絡中第二層或第三層的圖像?

謝謝。

+0

該secon d回答更合適,請重新選擇答案,以便找到更好更正確答案的人更容易 – Anuj

回答

13

同樣重量,你可以使用:

itorch.image(model:get(1).output) 
+0

謝謝!讓我嘗試一下。 – James

+0

正是我需要的,謝謝 – Anuj

5

形象化的權重:

-- visualizing weights 
n = nn.SpatialConvolution(1,64,16,16) 
itorch.image(n.weight) 

形象化特徵地圖:

-- initialize a simple conv layer 
n = nn.SpatialConvolution(1,16,12,12) 

-- push lena through net :) 
res = n:forward(image.rgb2y(image.lena())) 

-- res here is a 16x501x501 volume. We view it now as 16 separate sheets of size 1x501x501 using the :view function 
res = res:view(res:size(1), 1, res:size(2), res:size(3)) 
itorch.image(res) 

更多:https://github.com/torch/tutorials/blob/master/1_get_started.ipynb