Tensorflow：テンソルのチャンネルをRGBからBGRに切り替えるにはどうすればいいですか？

Question

画像をエンコードするテンソルをRGBからBGRに簡単に変換できますか？

ようなものになるだろうPythonで

：

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

Answer 1

あなたはtf.strided_sliceまたはtf.reverseを使用することができます。例えば

、tf.reverseのAPIはtensorflowのR1.0から変更されていることを

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]) 

注意。

Answer 2

チャネルが最後の次元であると仮定します。

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

Answer 3

tf.splitも使用できます。
スニペットからtensorflow-vgg16/blob/master/vgg16.py#L5：

red, green, blue = tf.split(3, 3, rgb_scaled) 
bgr = tf.concat(3, [blue, green, red])