我有5个波段的图像数据库:
RGB + alpha (a) + topography (t).
我知道如何反转RGB(无其他波段)以获得BGR:tf.reverse(image_patch, axis = [-1])
带有image_patch
:Tensor("Sub:0", shape=(256, 256, 5), dtype=float32)
但是如何按照以下顺序更改此张量的尺寸:BGRat
?
答案 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]]