channel_first 和channel_last

以三通道为例

1、channel_first (channel, height,width)

设上图为(3,2,2

解释:三个通道(3层)每个通道对应一个二维数组,即上图中的一层

黄:((1,2),

          (3,4))

绿:((5,6),

           (7,8))

蓝:((9,10),

           (11,12))

2、channel_last (height,width,channel)

上图为(2,2,3)

解释:先构建一个2*2的数组,数组中每个元素不是一个数值,而是3个数值,即三个通道的值。

(【1,5,9】,【2,6,10】)

(【3,7,11, 【4,8,12】)

3.总结

channel_first:先分层,每层一个二维数组,最终的array是包含三个二维数组的3维数组。

channel_last: 不分层,把三层看作一个整体,但是这个整体的每个元素不再是一个值,而是3个值。

猜你喜欢

转载自blog.csdn.net/qq_41033241/article/details/108339184