Ошибка при подаче изображений на график TensorFlow

Question

Я пытаюсь загрузить некоторые изображения в график TensorFlow, который равен RGB, однако мне хотелось бы, чтобы график преобразовывал их в оттенки серого перед обработкой.

x = tf.placeholder(tf.float32, shape=[None, 32, 32, 1], name='x') 

gray = tf.image.rgb_to_grayscale(x, name='grayscale') 

Однако я получаю ошибку

ValueError: Cannot feed value of shape (250, 32, 32, 3) for Tensor 'x:0', which has shape '(?, 32, 32, 1)' 

Я экспортировал notebook с ошибкой и загрузив его на Github как md файл для Комплектность и краткости.

Я понимаю, что ошибка связана с тем, что x_batch находится в форме RGB.

Однако я думал, что TensorFlow сделает преобразование автоматически.

С tf.image.rgb_to_grayscale обертывает входы, не должно TF делать grayscaling как часть сеанса?

Или я пропустил понимание того, как это работает?

Answer 1

Функция tf.image.rgb_to_graycale ожидает тензор входного сигнала с его последним измерением, имеющим размер 3. Для экземпляра партии изображений формы (250, 32, 32, 3) в вашем случае, или это может быть одно изображением формы (32, 32, 3).

Если вы хотите кормить RGB изображения и немедленно обработать их в оттенках серого, вы можете сделать:

images = tf.placeholder(tf.float32, shape=[None, 32, 32, 3]) 
gray_images = tf.image.rgb_to_grayscale(images) # has shape (None, 32, 32, 1)