图片卷积后的特征图可视化

import torch
import torch.nn as nn
import numpy as np
import matplotlib.pyplot as plt
import matplotlib
image = np.array(plt.imread('/Users/mac/Desktop/22222.jpeg'))
plt.imshow(image)
plt.show()
net = nn.Sequential(nn.Conv2d(3,3,11))
x = np.array(plt.imread('/Users/mac/Desktop/22222.jpeg'))
x = torch.from_numpy(x)    # 转换成张量
x = x.float()# 将它的数据类型转换成float32
x = x.permute(2,0,1) # 交换它的维度
x = x.unsqueeze(0) # 增加第0个维度
y = net(x)
y = y.permute([0,2,3,1])
plt.imshow(np.array(y[0].data))
plt.show()