hình ảnh Hình dung trong lớp trung gian trong ngọn đuốc (lua)

Question

Trong mô hình conv lưới, tôi biết làm thế nào để hình dung các bộ lọc, chúng ta có thể làm itorch.image (model: nhận được (1) .weight)

Nhưng làm thế nào thế nào tôi có thể trực quan hóa các hình ảnh đầu ra sau khi convolution? đặc biệt là những hình ảnh trong lớp thứ hai hoặc thứ ba trong một mạng lưới thần kinh sâu?

Cảm ơn.

Answer 1

Tương tự như trọng lượng, bạn có thể sử dụng:

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

Answer 2

Để hình dung trọng:

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

Để hình dung các bản đồ tính năng:

-- 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) 

Để biết thêm: https://github.com/torch/tutorials/blob/master/1_get_started.ipynb