Keras flow_from_directory ограничивает количество примеров

Question

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

Для контекста я хотел бы иметь возможность использовать небольшое подмножество общих изображений для целей тестирования без необходимости создания отдельного каталога верхнего уровня для меньшего набора данных, так как я извлекаю эти данные из блоков AWS S3 во время тренировки.

Answer 1

Создать keras.preprocessing.image.ImageDataGenerator с аргументом validation_split, указанным как float. В таком случае вы можете использовать аргумент subset в flow_from_directory, чтобы получить только несколько образцов из каждого каталога. Подробнее здесь .

Если вы хотите N изображений из каждой папки отдельно, вам нужно будет подсчитать, сколько файлов находится в каждом каталоге, и соответственно настроить разделение проверки поезда.