RGB를 회색조로 수동 변환 tensorflow

Question

tensorflow에서 라이브러리 사용없이 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() 

나는 변환을 위해 첫 번째 채널을 추출했습니다.

TypeError: Invalid dimensions for image data 

어떻게해야합니까 말을 imread 사용하는 동안 는하지만이 오류가 발생합니다? plt.imshow의 doc 가입일

Answer 1

(X) :

X : array_like, 형상 (N, M) 또는 (N, m, 3), (N, m, 4)

여기에 [없음, 없음, 1] 모양의 입력이 있습니다. 다음과 같이 마지막 치수를 제거하면됩니다.

result = np.squeeze(result, 2)