Я хочу обучить модель, используя некоторый набор изображений. У меня есть файл CSV со следующей информацией:
filename,width,height,class,xmin,ymin,xmax,ymax
xmin, ymin, xmax, ymax образуют область, которую необходимо обрезать. Путь к каталогу находится в отдельной переменной. Обучающие и проверочные изображения находятся в одном каталоге, у меня есть отдельный CSV-файл, в котором указано, какие файлы предназначены для проверки. В итоге я выполнил следующие шаги:
- Прочитать файл CSV (я использовал
pandas инструмент) - Путь сборки для каждого обучающего образа
- Загрузить каждое изображение в PIL object
- Обрезать каждое изображение
- ...
На этом я остановился, потому что не уверен, что делать с этими объектами PIL. Думаю, я мог бы сохранить их во временном каталоге, загрузить с помощью ImageDataGenerator.image_dataset_from_directory() и удалить, когда закончу. Но правильно ли это? Боюсь, для больших наборов данных это будет ужасно неэффективно. Может, я смогу все выполнить, используя только Керас? Могу ли я добавить изображения к объекту ImageDataGenerator в l oop? Я знаю, что есть инструменты для работы с CSV, но мои файлы содержат только часть данных.