如何将240 * 240 * 4的np数组打印或转换为240 * 240 * 1？

Question

我正在寻找只想打印np数组的第一个通道的东西。

原始尺寸= 240 * 240 * 4 目标尺寸= 240 * 240 * 1（仅第一个频道。

我在下面尝试过，但似乎确实可行。

image[:,:,:1]

但是将尺寸为240 * 240 * 1的文件保存回png或jpg无效

示例代码

import numpy as np
from PIL import Image
import scipy.misc as sp
image = np.array(Image.open("FLAIR-148.png"))
test_image = image[:,:,:1]
sp.imsave('out.png', test_image)

输出

File "/anaconda3/lib/python3.6/site-packages/scipy/misc/pilutil.py", line 327, in toimage
raise ValueError("'arr' does not have a suitable array shape for "
ValueError: 'arr' does not have a suitable array shape for any mode.

Answer 1

如果您不对最后一个索引进行切片（即image[:, :, 1]），那么一切应该正常：

import numpy as np
from PIL import Image
import scipy.misc as smc

image = np.array(Image.open("FLAIR-148.png"))
test_image = image[:, :, 1]
smc.imsave('out.png', test_image)

基本上，scipy.misc.imsave不知道如何处理形状为(M, N, 1)的数组。但是，它确实知道应该将形状为(M, N)的数组保存为灰度图像。

您可能还需要将数组转换为uint8以确保结果一致。这是一个完整的最小示例：

import scipy.misc as smc

# get the test image as an array
img = smc.face()

# slice test image
img = img[:, :, 1]

# convert to uint8
img = img.astype('uint8')

# save
smc.imsave('test.png', img)

输出：

注意

scipy.misc.imsave is deprecated。建议改用imageio.imwrite。