Tensorflow:如何将张量的通道从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])

注意,api tf.reverse是从tensorflow r1.0改变的.

Answer 2

假设通道是最后一个维度。

channels = tf.unstack (image, axis=-1)
image    = tf.stack   ([channels[2], channels[1], channels[0]], axis=-1)