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.

В то время как аналогичный код для чтения данных обученияработает правильно.Я проверил, существуют ли изображения по заданному пути, что они и делают.Каковы некоторые возможные причины этой ошибки?Как я могу попытаться отладить проблему?

РЕДАКТИРОВАТЬ: Это задача регрессии, поэтому все изображения находятся в одном каталоге, а не в подкаталогах, как и следовало ожидать для классификацииtask.

EDIT 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 как обычно работает.