如何扩展和填充黑白图像的第三维

Question

我有一个 (224,224) 形状的黑白图像，但我想要 (224,224,3)，所以我需要扩展 dim，但不是空值，所以 np.expand_dims 或np.atleast_3d 帮不了我。我怎样才能正确地做到这一点？谢谢

我用的是：

from PIL import Image
img = Image.open('data/'+link)
rsize = img.resize((224,224))
rsizeArr = np.asarray(rsize)

Answer 1

当我们使用 numpy.dstack() 时，我们不必手动扩展维度，它会处理这项工作并沿着我们想要的第三轴堆叠它。

In [4]: grayscale = np.random.random_sample((224,224))

# make it RGB by stacking the grayscale image along depth dimension 3 times
In [5]: rgb = np.dstack([grayscale]*3)

In [6]: rgb.shape
Out[6]: (224, 224, 3)

对于您的具体情况，应该是：

rsize_rgb = np.dstack([rsize]*3)

无论出于何种原因，如果您仍想将灰度图像的维度扩大1，然后将其制成RGB图像，那么您可以使用numpy.concatenate()如：

In [9]: rgb = np.concatenate([grayscale[..., np.newaxis]]*3, axis=2)
In [10]: rgb.shape
Out[10]: (224, 224, 3)

对于您的具体情况，它将是：

rsize_rgb = np.concatenate([rsize[..., np.newaxis]]*3, axis=2)

如何扩展和填充黑白图像的第三维

How to expand and fill third dim of black and white image

python

rgb

numpy

computer-vision

python-imaging-library