调用 tf.image.rgb_to_grayscale 后无法显示图像

Question

我正在开发人脸识别应用程序并准备读取 jpg 图像然后训练我的神经网络的数据集。为了提高准确性，我决定将图像转换为灰度（内部映射功能）。 这是我将图像转换为张量的方式：

from PIL import Image

image_data = tf.io.decode_jpeg(
  image_bytes_string,
  channels=3
)

image_data = tf.reshape(image_data , (277, 370, 3))

label = parsed['label']
label = tf.reshape(label, (1,))

return image_data, label

我如何显示图像的代码：

image_data = dataset_item[0].numpy()
img = Image.fromarray(image_data)
img

更改代码转换为灰度后，显示图像的代码显示此错误：

KeyError: ((1, 1, 1), '|u1')
During handling of the above exception, another exception occurred:
Cannot handle this data type

灰度代码：

image_data = tf.io.decode_jpeg(
  image_bytes_string,
  channels=3
)

grayscale = tf.image.rgb_to_grayscale(
  image_data
)

image_data = tf.reshape(grayscale, (277, 370, 1))

label = parsed['label']
label = tf.reshape(label, (1,))

return image_data, label

我如何将图像转换为灰度或 PIL 库有问题吗？

更新： 这是带有示例的 colab：https://colab.research.google.com/drive/1v72_C5i8HZzSLEy_p6d45EWq1byoF3fe

Answer 1

我确实陷入了同样的错误，但以下代码可以帮助您显示 gray_scale 图像。

在下面的代码中，您可以看到指标 [...,0] 被称为 Ellipsis_object。

grey_img = tf.image.rgb_to_grayscale(jpeg_image)
_ = plt.imshow(grey_img[...,0], cmap='gray')