python - Keras flow_from_dataframe 给出 0 个图像

Question

我正在尝试使用 flow_from_dataframe Keras读取训练和测试图像的方法。

我的训练和测试图像都在同一目录中，并且我从两个不同的 csv 文件读取路径。

我用于读取测试图像的代码如下所示，

# Read test file
testdf = pd.read_csv("test.csv")

# load images
test_datagen = ImageDataGenerator(rescale=1./255)

test_generator = test_datagen.flow_from_dataframe(
    dataframe=testdf, directory=IMAGE_PATH,
    x_col='image_name', y_col=None,
    has_ext=True, target_size=(10,10)
    ,batch_size=32,color_mode='rgb',shuffle=False, class_mode=None)

我得到这样的输出

Found 0 images.

虽然用于读取训练数据的类似代码可以正常工作。我检查了图像是否存在于给定路径中，它们确实存在。导致此错误的可能原因有哪些？我该如何尝试调试该问题？

编辑:这是一项回归任务，因此所有图像都位于单个目录中，而不是像分类任务那样位于子目录中。

编辑2:我将usecols=[0]添加到read_csv，现在test_datagen找到目录中的所有图像，而不仅仅是中提到的图像test.csv 文件

Answer 1

出现此问题的原因是数据框中存在 NaN。忽略这些列是行不通的。解决方案是用其他内容替换 NaN。例如，

testdf = pd.read_csv("test.csv")
testdf.fillna(0, inplace=True)

这会将 NaN 替换为 0。然后照常使用 ImageDataGenerator。