CODEWITHSUNDEEP
luaneural-networktorchconv-neural-network
Ethan
Ethan

Reputation: 1266

Visualize Neural Net Layers in Torch, without itorch

I have a neural net with several convolution layers and I'd like to visualize the feature maps I'm creating.

I've seen this post: Visualize images in intermediate layers in torch (lua)

which suggests using itorch, but it requires running my code in an itorch notebook which I would like to avoid.

Are there any other Torch packages which can be used to visualize convolution layers?

EDIT (with detailed solution):

Since I found so few resources online about how to go about doing this I documented my full solution and slapped it on Github. Anyone who wants to visualize neural nets in torch can just go here to get started!

https://github.com/egaebel/torch-neural-network-visualization

Much thanks again to YuTse for the gnuplot tip!

Upvotes: 3

Views: 1574

Answers (1)

yutseho
yutseho

Reputation: 1689

gnuplot

in itorch console mode(itorch) / torch mode(th)

require 'image';

a = image.lena();

require 'gnuplot';

gnuplot.figure(1);

gnuplot.imagesc(a[1])

Upvotes: 0

Related Questions