Flow_from_dataframe无法正常工作

我在使用flow_from_dataframe时已发出。 我使用的代码是:

train_data_generator = train_data_generator.flow_from_dataframe(
        train_dataframe,
        IMG_DIRECTORY,
        x_col="Filename",
        y_col="Label",
        target_size=RAW_IMG_SIZE,
        batch_size=BATCH_SIZE,
        classes=CLASSES,
        class_mode='categorical')

但是,在运行时,我得到以下信息:

找到0个经过验证的图像文件名,它们属于9个类。 找到0个经过验证的图像文件名,它们属于9个类。 找到0个经过验证的图像文件名,它们属于9个类。

我看到应该创建一些子目录,但是指定类还不够吗?

编辑:此外,在训练模型时,在每个时期,它都表示损失:nan-categorical_accuracy:nan。我想这是因为没有像我之前所说的那样检测到图像。有人可以帮忙吗?

谢谢!

评论