我正在运行此代码,但无法克服此错误。我在Google驱动器上使用的是zip格式的数据集,并且所有文件夹都具有相同的名称,没有不同格式的文件(均为jpeg),也没有任何其他类型的文件。
image_gen = ImageDataGenerator(
rescale=1./255
)
train_data_gen = image_gen.flow_from_directory(
#batch_size=batch_size,
directory=train_dir,
shuffle=True,
target_size=(IMG_HEIGHT, IMG_WIDTH),
class_mode='categorical')
sample_training_images, _ = next(train_data_gen)