问题描述
在使用数据生成器时,我只想加载没有标签的图像。我正在训练用于在imagenet数据集上进行图像重建的自动编码器。 Imagenet数据集的架构与下面给出的类似。
数据/ 数据/火车/ 数据/火车/红色/ 数据/火车/蓝色/ 数据/火车/绿色/ 数据/火车/粉红色/
我正在使用下面给出的代码。
train_it = datagen.flow_from_directory(directory=”/dataset/Imagenet2012/train/”,target_size=(224,224),color_mode=”rgb”,batch_size=32,class_mode=None,shuffle=True,seed=42
)
我已将class_mode更改为None,但是,当我调用fit函数时,出现以下错误。适合功能的代码如下。
autoencoder.fit(val_it,val_it,epochs=5,verbose=2,shuffle=True
)
我得到的错误是
y argument is not supported when data is a generator or Sequence instance.Instead pass targets
as the second element of the generator.
我想将原始图像作为y变量传递给model.fit函数。该数据集包括用于训练的1281167张图像。当我尝试使用numpy加载所有图像时,加载大约需要6个小时。 请帮我解决一下这个。 或者,请为我建议一个用于自定义数据生成器的代码。
谢谢
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)