c - 如何使用手电筒添加图像 channel ？

我已加载3个通道的jpg图像，如何添加最后一个通道以生成4个通道的图像张量？

最佳答案

假设您有一个包含3个通道的图像，高度height和宽度width：

a = torch.Tensor(3, height, width) -- this is your image
b = torch.Tensor(1, height, width) -- the channel you want to add
c = torch.cat(a,b,1)

一个工作示例：

th> a = torch.Tensor(3,3,3):fill(1)
th> b = torch.Tensor(1,3,3):fill(0)
th> c = torch.cat(a,b,1)
th> c
(1,.,.) =
  1  1  1
  1  1  1
  1  1  1

(2,.,.) =
  1  1  1
  1  1  1
  1  1  1

(3,.,.) =
  1  1  1
  1  1  1
  1  1  1

(4,.,.) =
  0  0  0
  0  0  0
  0  0  0
[torch.DoubleTensor of size 4x3x3]

关于c - 如何使用手电筒添加图像 channel ？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/41607160/