Tensorflow:如何将张量的 channel 从 RGB 切换到 BGR？

Question

是否有一种快速方法将编码图像的张量从 RGB 转换为 BGR？

在 python 中，这将是这样的:

image = image[:, :, [2,1,0]]

Answer 1

您可以使用tf.strided_slice或tf.reverse .

例如，

import tensorflow as tf

img = tf.reshape(tf.range(30), [2, 5, 3])

# the following two lines produce equivalent result:
img_channel_swap = img[..., ::-1]
img_channel_swap_1 = tf.reverse(img, axis=[-1])

请注意，tf.reverse的api是从tensorflow r1.0更改的。