Как изменить порядок измерения изображения

Question

У меня есть база изображений с 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?

Answer 1

Вы можете использовать метод 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]]