我通过使用fit_genertor解决了这个问题。我在这里找到了解决方案。我应用了@ Dat-
Nguyen的解决方案。
您只需创建两个迭代器,一个用于训练,一个用于验证,然后创建您自己的生成器,即可从数据集中提取批次,并以(batch_data,batch_labels)的形式提供数据。最后,在model.fit_generator中,您将传递train_generator和validation_generator。
我通过使用fit_genertor解决了这个问题。我在这里找到了解决方案。我应用了@ Dat-
Nguyen的解决方案。
您只需创建两个迭代器,一个用于训练,一个用于验证,然后创建您自己的生成器,即可从数据集中提取批次,并以(batch_data,batch_labels)的形式提供数据。最后,在model.fit_generator中,您将传递train_generator和validation_generator。