NHWC与NCHW

  • "channels_last":设置为 “NHWC” 时,排列顺序为 [batch, height, width, channels]
  • "channels_first":设置为 “NCHW” 时,排列顺序为 [batch, channels, height, width]
    其中 N 表示这批图像有几张,H 表示图像在竖直方向有多少像素,W 表示水平方向像素数,C 表示通道数。
  • Tensorflow默认为 "NHWC"
  • 假设RGB三通道两种格式的区别如下图所示:
    在这里插入图片描述

(1)如果通道在最低维度0[channel, height, width], “NHWC” ,RGB三颜色分成三组,在第一维度上找到三个RGB颜色
(2)如果通道在最高维度2[height, width, channel], “NHWC” ,在第三维度上找到RGB三个颜色

例:假设1, 2, 3, 4-红色 5, 6, 7, 8-绿色 9, 10, 11, 12-蓝色

In [1]: import tensorflow as tf
In [2]: tf.InteractiveSession()
Out[2]: <tensorflow.python.client.session.InteractiveSession at 0x1b8601ddb70>

In [3]: a = tf.reshape([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12], [3, 2, 2]).eval()
In [4]: a
Out[4]:
array([[[ 1,  2],
        [ 3,  4]],

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

       [[ 9, 10],
        [11, 12]]])
        
In [5]: a1 = tf.transpose(a, [1, 2, 0]).eval()
In [6]: a1
Out[6]:
array([[[ 1,  5,  9],
        [ 2,  6, 10]],

       [[ 3,  7, 11],
        [ 4,  8, 12]]])
In [8]: b = tf.reshape([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12], [2, 2, 3]).eval()  # 错误

In [9]: b
Out[9]:
array([[[ 1,  2,  3],
        [ 4,  5,  6]],

       [[ 7,  8,  9],
        [10, 11, 12]]])
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值