在火炬(lua)中可视化中间层中的图像

时间:2015-07-20 06:18:54

标签: image lua data-visualization convolution torch

在conv-nets模型中,我知道如何可视化过滤器,我们可以做itorch.image(model:get(1).weight)

但是如何在卷积后有效地可视化输出图像?特别是深层神经网络中第二层或第三层的图像?

感谢。

2 个答案:

答案 0 :(得分:13)

与体重相似,您可以使用:

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

答案 1 :(得分: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