Answers for "visualizing of convolutional kernels using pytorch"

0

visualizing of convolutional kernels using pytorch

conv1 = nn.Conv2d(3, 1, 3)
weight = conv1.weight.data.numpy()
plt.imshow(weight[0, ...])
Posted by: Guest on December-18-2020

Python Answers by Framework

Browse Popular Code Answers by Language