Python 3.x 可视化要素地图

有人知道为什么我的代码在验证过程中使用时以灰度显示功能映射吗

输出如下所示：

我期待着下面这样的东西（我的意思不是灰度），但不知道怎么做：

这是我的密码

def vis_tensor(data):
    data  =data.data.cpu().numpy()
    data =data[0].transpose((1, 2, 0))
    data = data - data.min()
    data = data / data.max()
    data = data * 255
    mean = np.repeat(data.mean(-1, keepdims=True),3 ,2)
    mean = mean.astype('uint8')
    pylab.imshow(mean)
    pylab.savefig('mean.png')
    max = np.repeat(data.max(-1, keepdims=True), 3, 2)
    max = max.astype('uint8')
    pylab.imshow(max, cmap='plasma')
    pylab.savefig('max.png')

---

您在准备培训和测试数据集时是否使用了autoencoder？您是否可以尝试删除

np，重复（…

行）？我认为您创建的像素具有相同颜色值的三倍（如rgb 128128）它总是会产生灰色阴影。移除它，我认为只要你设置正确的cmap，它就会工作。嘿@Anton，我已经移除了你说的线，但我面临图像数据错误的无效维度！有什么想法吗？是的，pylab期望每像素3个rgb值。尝试使用matplotlib绘制数据，这也将允许你轻松更改把颜色涂成灰色。