如何将灰度图像的张量转换为3通道图像?

Question

我有一个大小为 W x H 的 n 个灰度图像的张量： images.shape 输出 (n， W， H， 1( 我想通过添加灰度通道的两个副本(最后一个维度(将这个张量转换为 (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]]]])