tensorflow里面把彩图灰度转换的函数是tf.image.rgb_to_grayscale,这个函数关键的参数就一个,那就是输入图像。
例如:
import matplotlib.pyplot as plt;
import tensorflow as tf;
import numpy as np;
with tf.Session() as sess:
image_raw_data_jpg = tf.gfile.FastGFile('11.jpg', 'r').read()
image_data = tf.image.decode_jpeg(image_raw_data_jpg)
image_data = sess.run(tf.image.rgb_to_grayscale(image_data))
print image_data.shape
plt.imshow(image_data[:,:,0],cmap ='gray')
plt.show()
结果: