Как загрузить изображения с метками классов из одного каталога в модель Keras, используя фрейм данных, содержащий имена файлов и метки?

Question

У меня есть две отдельные папки "train_images" и "val_images".У меня есть все изображения внутри этих двух папок.Я делаю категориальную классификацию, используя модель DenseNet121.У меня есть два разных y_df (датафрейма) для обучения и проверки.Я знаю о flow_from_directory, но для этого мне нужно создавать разные подформаторы внутри каждого "train_images" и "val_images" на основе классов.

Я хочу знать, не хочу ли я создавать этих подзаголовкови не используйте flow_from_directory, как мне передать эти изображения в модель?Я использую Keras с Tensorflow в качестве бэкэнда.

Answer 1

Если у вас есть фрейм данных, который содержит имена файлов изображений вместе с их метками, вы можете использовать недавно введенный метод flow_from_dataframe(), который был точно определен для этой цели.Он принимает кадр данных Pandas, который содержит имена файлов изображений в одном столбце и соответствующие им метки в другом столбце (столбцах). Он еще не был официально выпущен, поэтому вам необходимо использовать последнюю версию Keras для разработки в Keras Github репозитории .Ожидается, что он будет в следующем выпуске.

Обновление: Он был включен в Предварительная обработка Keras 1.0.4 и Keras2.2.3 .

Также есть учебник об этом методе.