Как загрузить изображения для классификации проблемы с помощью Keras

Question

Я работаю над проблемой классификации изображений с использованием фреймворка Keras. Это проблема двоичной классификации, и у меня есть 2 папки обучающий набор и тестовый набор , который содержит изображения обоих классов. У меня нет отдельной папки для каждого класса (скажем, кошка против собаки). Keras ImageDataGenerator работает, когда у нас есть отдельные папки для каждого класса (папка cat и папка dog). Но у меня есть все изображения в одной папке обучающий набор , и я не понимаю, как действовать дальше. Пожалуйста, предложите, как загрузить изображения.

У меня также есть 2 файла CSV - train.csv и test.csv. Файл train.csv содержит 2 столбца, а именно image_id и class_name. test.csv содержит image_id. Обратите внимание, что image_id совпадает с именем файлов в папках изображений.

Answer 1

В последних версиях ImageDataGenerator есть метод с именем flow_from_dataframe, который делает именно то, что вы хотите.

Обычно он используется для первой загрузки CSV-файла в pandas DataFrame, создания экземпляра ImageDataGenerator, а затем для вызова flow_from_dataframe с тремя важными параметрами:

• directory: папка, в которой хранятся ваши данные.
• x_col: столбец в DataFrame, который содержит имена файлов внутри directory, которые соответствуют вашим данным обучения / тестирования.
• y_col: столбец в DataFrame, соответствующий меткам, которые будут выводиться генератором.

Затем вы используете этот генератор как любой другой, вызывая fit_generator. Более подробная информация и примеры доступны здесь .