Блог

Я использую Autokeras для классификации изображений RGB в Google Colab. Мой каталог выглядит следующим образом:

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

Я загружаю изображение, используя tf.data.Набор данных следующим образом:

 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 должен быть размер пакета.

Интересно, что вызывает это и как я должен это исправить?