tensorflow RGB三通道图转换为灰度图

最近开始写一些深度学习CV方向相关的文章记录一下平时遇到的问题。

tensorflow中内置了将三通道彩色图片转换为灰度图的方法，使用tf.image.rgb_to_grayscale()，需要注意的是tensorflow默认三通道为R,G,B,而opencv默认为B,G,R。

import tensorflow as tf;  
import numpy as np;
import cv2
 
with tf.Session() as sess:

	with tf.gfile.FastGFile('test.jpg', 'rb') as f:
        image = f.read()
	im = tf.image.decode_jpeg(image)
	im = sess.run(tf.image.rgb_to_grayscale(im))
	print (im.shape)
	cv2.imshow("gray_window", im)
	cv2.waitKey()