Asked to retrieve element 0, but the Sequence has length 0 解决

最新推荐文章于 2024-03-06 21:00:46 发布

arenascat

最新推荐文章于 2024-03-06 21:00:46 发布

阅读量1.1w

点赞数 16

分类专栏： AI 日常问题文章标签：图片识别 Keras

本文链接：https://blog.csdn.net/u012388993/article/details/113789034

版权

日常问题同时被 2 个专栏收录

43 篇文章

订阅专栏

15 篇文章

订阅专栏

在尝试使用Keras进行图像分类时遇到fit_generator报错，提示找不到图片。问题在于目录结构不正确和fit_generator的弃用。解决方案是更新为model.fit，并确保训练数据目录符合标准格式，将图片按类别放入子文件夹。通过调整目录结构并使用model.fit，成功避免错误并完成模型训练。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题现象

这个问题是这样的，在用Keras做一个极为简单的图像分类时候，我使用ImageDataGenerator构建了图片的预处理模型，用函数flow_from_directory来获取文件夹中的所有图片，然后使用函数fit_generator 去读取，这个时候出现model.fit_generator(train_generator, 这个位置报错，提示Asked to retrieve element 0, but the Sequence has length 0

部分代码如下：

train_datagen = ImageDataGenerator(rescale=1./255,
                                   shear_range=0.7,
                                   zoom_range=0.7,
                                   horizontal_flip=True)

test_datagen = ImageDataGenerator(rescale=1./255)
train_generator = train_datagen.flow_from_directory(train_path,
                                                    target_size=(64,64),
                                                    batch_size=6,
                                                    class_mode='binary')

validation_generator = test_datagen.flow_from_directory(test_path,
                                                        target_size=(64,64),
                                                        batch_size=6,
                                                        class_mode='binary')

# 这里报错
model.fit_generator(train_generator,
                    steps_per_epoch=18,   #训练数据的数量除以batch_size，108/6=18
                    epochs=200,   #训练迭代次数
                    validation_data=validation_generator,
                    validation_steps=200)