当使用数据集作为输入时，不支持TensorFlow`Y`参数

Question

当我运行此代码以生成数据集和训练gan时，

batch_size = 32
img_height = 128
img_width = 128

train_ds = tf.keras.utils.image_dataset_from_directory(
  data_dir,
  validation_split=0.2,
  subset="training",
  seed=123,
  image_size=(img_height, img_width),
  batch_size=batch_size)

val_ds = tf.keras.utils.image_dataset_from_directory(
  data_dir,
  validation_split=0.2,
  subset="validation",
  seed=123,
  image_size=(img_height, img_width),
  batch_size=batch_size)

autoencoder.fit(train, train,
                epochs=10,
                shuffle=True,
                validation_data=val)

它从我所看到的，我需要使输入元组返回此错误

`y` argument is not supported when using dataset as input.

，但我不确定如何做到这一点，我找不到任何东西那告诉我什么。

Answer 1

默认情况下，（图像，标签）的元组。这就是为什么接受 y 参数为标签应直接来自 DataSet 。

如果要手动为数据集指定标签，请将其作为参数 labels 在调用 image_dataset_from_directory 时将其传递。注意

应根据图像文件路径的字母数字对标签进行排序（通过python中的 os.walk（Directory）获得）。