Как передать файлы изображений (jpg), представленные в каталогах своих классов, в Оценщик Tensorflow для обучения?

Question

Структура каталогов:

Data
-Cats
--<images>.jpg
-Dogs
--<images>.jpg

Я тренирую (n-ary) модель классификации. Я хочу создать input_fn для обслуживания этих изображений для обучения.

Размеры изображения (200, 200, 3). У меня есть генератор (керас) для них, если они могут быть использованы как-то.

Я долго искал, но не нашел простого способа сделать это. Я думал, что это должен быть стандартный вариант использования? например, Keras предоставляет flow_from_directory для обслуживания моделей keras. Мне нужно использовать tf.estimator для AWS Sagemaker, поэтому я застрял с этим.

Answer 1

Используя модуль tf dataset Module, вы можете передавать свои данные непосредственно в оценщик. В основном у вас есть 3 способа интегрировать это в ваш API: 1. преобразовать ваши изображения в tfrecords и использовать tfrecorddataset2 использовать набор данных tf из генератора для использования генераторов 3 попытаться внедрить эти функции декодера в ваш inputpipeline