Я создал (мою первую) сверточную нейронную сеть в кератах и теперь хотел бы запустить ее с использованием выборочных весов. Изображения находятся в соответствующих папках по классам, а веса хранятся в словаре с именем файла изображения в качестве ключа, потому что это кажется логичным способом ссылки на них. Я использую приведенный ниже код в качестве генератора для загрузки изображений и классификации:
# how my training images are loaded
train_batches = ImageDataGenerator().flow_from_directory('path_to_train_folder',
target_size=(img_vert, image_horz),
classes = ['class_1','class_2'],
batch_size=32)
Итак, я понимаю, что генератор flow_from_directory возвращает пакеты (img, class), и мне нужно, чтобы он возвращал (img, class, sample_weight) для каждого изображения в пакете. Я считаю, что могу передать такой генератор в метод .fit_generator так же, как и в предыдущей версии.
Есть ли простой способ использовать ImageDataGenerator (). Flow_from_directory для возврата значений дополнительного веса в пакетах?
У меня сложилось впечатление, что мне нужно будет создать функцию генератора, которая будет делать то, что я хочу, но я не уверен, как вернуть пакет в соответствующем объекте итератора каталога предварительной обработки keras (и я только смутно понимаю, что только что набрал) , Если это необходимо сделать с нуля, любые советы о том, с чего начать, приветствуются!
Спасибо!