python - 如何将灰度图像的张量转换为3 channel 图像？

Question

我有一个由 n 个灰度图像组成的张量，尺寸为 W x H:images.shape 输出 (n, W, H, 1)我想通过添加灰度 channel (最后一个维度)的两个副本来将此张量转换为 (n, W, H, 3)。

如果有人提出一个好方法来做到这一点，我将不胜感激。

谢谢!

Answer 1

据我了解，您正在寻找类似的东西(根据 numpy 标签判断):

>>> import numpy as np
>>> x = np.array([[[1,5,7],[9,2,1]], [[5,6,7], [8,2,4]]])
>>> x
array([[[1, 5, 7],
        [9, 2, 1]],

       [[5, 6, 7],
        [8, 2, 4]]])
>>> np.stack((x,) * 3, axis=-1)
array([[[[1, 1, 1],
         [5, 5, 5],
         [7, 7, 7]],

        [[9, 9, 9],
         [2, 2, 2],
         [1, 1, 1]]],


       [[[5, 5, 5],
         [6, 6, 6],
         [7, 7, 7]],

        [[8, 8, 8],
         [2, 2, 2],
         [4, 4, 4]]]])