如何使用matplotlib/numpy将数组保存为灰度图像？

14 浏览
0 Comments

如何使用matplotlib/numpy将数组保存为灰度图像？

我试图将一个128x128像素的numpy数组保存为灰度图像。

我本以为pyplot.imsave函数可以完成这个任务，但事实并非如此，它将我的数组转换为RGB图像。

我尝试在转换过程中强制使用灰度映射，但保存的图像虽然呈现为灰度，但仍具有128x128x4的尺寸。

下面是我写的一个代码示例来展示这种行为：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.image as mplimg
from matplotlib import cm
x_tot = 10e-3
nx = 128
x = np.arange(-x_tot/2, x_tot/2, x_tot/nx)
[X, Y] = np.meshgrid(x,x)
R = np.sqrt(X**2 + Y**2)
diam = 5e-3
I = np.exp(-2*(2*R/diam)**4)
plt.figure()
plt.imshow(I, extent = [-x_tot/2, x_tot/2, -x_tot/2, x_tot/2])
print I.shape
plt.imsave('image.png', I)
I2 = plt.imread('image.png')
print I2.shape
mplimg.imsave('image2.png',np.uint8(I), cmap = cm.gray)
testImg = plt.imread('image2.png')
print testImg.shape

无论哪种情况，"print"函数的结果都是(128,128,4)。

有人能解释一下为什么imsave函数会创建这些维度，即使我的输入数组是亮度类型吗？

当然，有人有办法将数组保存为标准的灰度格式吗？

谢谢！

0