卷积神经网络可视化
conv1_weights = list(model_saved.parameters())[0]
conv1_images = make_grid(conv1_weights, normalize=True).cpu()
plt.figure(figsize=(8, 8))
plt.imshow(conv1_images.permute(1, 2, 0).numpy())
输出:可视化了第一个卷积层的卷积核
原创
2021-09-06 18:13:24 ·
538 阅读 ·
0 评论