python - 复制/ reshape 高维 numpy 数组

Question

我有一个形状像这样的 numpy 数组

x.shape
(100, 1, 300, 300)

将此视为对大小为 300x300 的灰度图像的 100 次观察。灰度图像只有 1 个 channel ，因此形状中的第二个 channel 为 1。

我想将其转换为具有 3 个 channel 的 RGB 图像数组。我只想将灰度图像复制到其他两个 channel 。

所以最终的形状是 (100, 3, 300, 300)

我该怎么做？

Answer 1

使用np.repeat -

np.repeat(x,3,axis=1)

sample 运行-

In [8]: x = np.random.randint(11,99,(2,1,3,4))

In [9]: np.repeat(x,3,axis=1).shape
Out[9]: (2, 3, 3, 4)