TensorFlow的图像NCHW与NHWC

171 阅读 0 评论 113 点赞

我是靠谱客的博主娇气镜子，这篇文章主要介绍TensorFlow的图像NCHW与NHWC，现在分享给大家，希望可以做个参考。

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import tensorflow as tf

x = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]

with tf.Session() as sess:
    a = tf.reshape(x, [2, 2, 3])
    a = sess.run(a)
    print(a)
    print("----------------------------")
    b = tf.reshape(a,[3,2,2])
    b = sess.run(b)
    print(b)
    print("-----------------------------")
    c = tf.transpose(b,[1,2,0])
    c = sess.run(c)
    print(c)

结果：
[[[ 1  2  3]
  [ 4  5  6]]

 [[ 7  8  9]
  [10 11 12]]]
----------------------------
[[[ 1  2]
  [ 3  4]]

 [[ 5  6]
  [ 7  8]]

 [[ 9 10]
  [11 12]]]
-----------------------------
[[[ 1  5  9]
  [ 2  6 10]]

 [[ 3  7 11]
  [ 4  8 12]]]