tensorflow - Autokeras：元组索引超出范围

Question

我在 Google Colab 上使用 Autokeras 进行 RGB 图像分类。我的目录如下所示：

train
---class1
------img1.jpg
------img2.jpg
---class2
------img3.jpg
------img4.jpg

我使用 tf.data.Dataset 加载图像，如下所示：

train_ds = tfk.preprocessing.image_dataset_from_directory("/content/train",
                                                          image_size=(300,300))

然后我导入 Autokeras 并像这样使用它：

import autokeras as ak

clf = ak.ImageClassifier(objective="val_accuracy")
clf.fit(train_ds)

但这给了我错误：

IndexError: tuple index out of range

我使用element_spec，结果如下：

(TensorSpec(shape=(None, 300, 300, 3), dtype=tf.float32, name=None),
 TensorSpec(shape=(None,), dtype=tf.int32, name=None))

None批量大小应该在哪里。

我想知道是什么原因造成的，我应该如何解决这个问题？

标签： tensorflowkerastensorflow2.0auto-keras