问题描述
我正在使用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 (将#修改为@)