Преобразование RGB в оттенки серого вручную tensorflow

Question

Я хотел преобразовать изображение RGB в оттенки серого вручную без использования библиотеки в тензорном потоке. Поэтому я написал следующее ...

import tensorflow as tf 
import matplotlib.image as mpimg 
import matplotlib.pyplot as plt 

# First, load the image again 
filename = "MarshOrchid.jpg" 
raw_image_data = mpimg.imread(filename) 

image = tf.placeholder("float", [None, None, 3]) 
slice = tf.slice(image,[0,0,0],[-1,-1,1]) 

with tf.Session() as session: 
    result = session.run(slice, feed_dict={image: raw_image_data}) 
    plt.imshow(result) 
    plt.show() 

Я извлек первый канал изображения для преобразования. Но это порождает ошибку при использовании imread говоря

TypeError: Invalid dimensions for image data 

Что я должен делать?

Answer 1

Из doc из plt.imshow (Х):

X: array_like, форма (п, т) или (п, м, 3) или (п, м, 4)

Здесь у вас есть ввод формы [None, None, 1]. Вам просто нужно удалить последние измерения следующим образом:

result = np.squeeze(result, 2)