小编Wil*_*man的帖子

我有一组灰度 png 图像，分为 2 个目录。根据文档，我已使用 image_dataset_from_directory 将它们作为数据集对象加载。当我使用 element_spec 检查已加载的内容时，它说图像有 3 个通道：

from tensorflow.keras.preprocessing import image_dataset_from_directory
Dataset = image_dataset_from_directory('path/to/files')
Dataset.element_spec

返回：

找到属于 2 个类别的 14000 个文件。

(TensorSpec(形状=(无, 256, 256, 3), dtype=tf.float32, 名称=无), TensorSpec(形状=(无,), dtype=tf.int32, 名称=无))

使用 MATLAB 将图像保存为灰度 png，并且我已使用 Linux 命令文件确认它们是灰度的：

$ file path/to/files/class_1/file_1.png

path/to/files/class_1/file_1.png：PNG 图像数据，256 x 256，8 位灰度，逐行扫描

所以现在我要么需要告诉 image_dataset_from_directory 将这些文件加载​​为灰度，要么需要将 3 通道张量 Dataset 对象转换为 1 通道张量。我该怎么办呢？

编辑：

有关使用识别的磁盘上文件的更多信息（来自 ImageMagick）：

$ identify -verbose path/to/files/class_1/file_1.png

Image: AI_Optrap/Samples/Set4/relaxed/HL60_normoxia_1_1.png
  Format: PNG (Portable Network Graphics)
  Mime type: image/png
  Class: PseudoClass
  Geometry: 256x256+0+0
  Units: Undefined …