我在使用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。我想这是因为没有像我之前所说的那样检测到图像。有人可以帮忙吗?
谢谢!