如何解决Autokeras:元组索引超出范围
我正在使用Autokeras在Google Colab上进行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,3),dtype=tf.float32,name=None),TensorSpec(shape=(None,),dtype=tf.int32,name=None))
其中None
应该是批量大小。
我想知道是什么原因造成的,应该如何解决?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。