Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

visualizing of convolutional kernels using pytorch

conv1 = nn.Conv2d(3, 1, 3)
weight = conv1.weight.data.numpy()
plt.imshow(weight[0, ...])
Source by discuss.pytorch.org #
 
PREVIOUS NEXT
Tagged: #visualizing #convolutional #kernels #pytorch
ADD COMMENT
Topic
Name
7+7 =