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