利用np.stack()把不同1 channel的灰度图像合并为3 channel的RGB图

np.stack文档解释:


Join a sequence of arrays along a new axis.

The axis parameter specifies the index of the new axis in the dimensions of the result. For example, if axis=0 it will be the first dimension and if axis=-1 it will be the last dimension.

产生一个给定axis的新轴 ,沿着该轴堆砌数组

下面是代码测试:

a = np.arange(8).reshape((2,2,2))
array([[[0, 1],
        [2, 3]],

       [[4, 5],
        [6, 7]]])
b = np.arange(8).reshape((2,2,2))+1
array([[[1, 2],
        [3, 4]],

       [[5, 6],
        [7, 8]]])

第一个2是2个样本,后面2*2是像素矩阵,要将a,b变成2*2*2*2,第4个2表示颜色通道:

c = np.stack((a,b),axis=3)
array([[[[0, 1],
         [1, 2]],

        [[2, 3],
         [3, 4]]],


       [[[4, 5],
         [5, 6]],

        [[6, 7],
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值