“Visualisation des grains convolutionnels à l'aide de pytorch” Réponses codées

Visualisation des grains convolutionnels à l'aide de pytorch

conv1 = nn.Conv2d(3, 1, 3)
weight = conv1.weight.data.numpy()
plt.imshow(weight[0, ...])