如何更改图像尺寸的顺序

时间:2019-05-27 16:33:58

标签: python image tensorflow multidimensional-array

我有5个波段的图像数据库:

RGB + alpha (a) + topography (t).

我知道如何反转RGB(无其他波段)以获得BGR:tf.reverse(image_patch, axis = [-1])

带有image_patchTensor("Sub:0", shape=(256, 256, 5), dtype=float32)

但是如何按照以下顺序更改此张量的尺寸:BGRat

1 个答案:

答案 0 :(得分:0)

您可以使用tf.reverse_sequence()方法。

示例代码

import tensorflow as tf

# Parameters
rows = 2
shift = 3

a = tf.placeholder(dtype=tf.int64, shape=(rows, 5))
b = tf.reverse_sequence(a, seq_lengths=[shift] * rows, batch_dim = 0, seq_dim=1)

sess = tf.Session()
print(sess.run(b, feed_dict={a: [[1, 2, 3, 4, 5], 
                                [1, 2, 3, 4, 5]]}))

结果

# a - sample input
[[1 2 3 4 5]
 [1 2 3 4 5]]

# b - shifted columns
[[3 2 1 4 5]
 [3 2 1 4 5]]