python - 如何在 python 中将黑白图像转换为 3 维数组？

Question

我有 RGB 格式或灰度格式的图像(我通过 Gimp 转换它，比方说)，现在每次我以灰度加载图像，或者只是将其转换为灰度格式时，形状总是显示 [height, width]没有三维(颜色 channel 数)。

我知道通常黑白图像以这种格式存储，但我特别需要 [height, width, 1] 图像形状，你会得到的那个，比方说:

numpy.zeros(shape=[400, 400, 1])

Answer 1

您始终可以使用 np.expand_dims 添加“空”维度:

>>> a2d = np.ones((100, 200))
>>> a3d = np.expand_dims(a2d, axis=2)
>>> a3d.shape
(100, 200, 1)

或者使用 None 或 np.newaxis 进行切片:

>>> a2d[..., None].shape  # instead of "..." (Ellipsis) you could also use `[:, :, None]`
(100, 200, 1)

我更喜欢 np.expand_dims，因为它比切片更明确一点。

---

如果你有条件地需要它，首先检查arr.ndim:

if arr.ndim == 2:
    arr = np.expand_dims(arr, axis=2)