Потоковая передача изображений из каталога и связывание прогноза с именем файла в tenorflow

Question

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

Answer 1

Что вам нужно сделать, это разделить изображения в другую папку, соответствующую классу. Имя папки должно быть именем класса, с помощью ImageDataGenerator.flow_from_directory() Keras автоматически выведет имена классов на основе каталогов. Например, у вас должна быть папка с именем «data», которая содержит 2 папки с именами «cat» и «dog».

Затем вы можете вызвать метод ImageDataGenerator.flow_from_directory("path/to/folder/data"), и Keras создаст набор данных с двумя классами "cat" и "dog".

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