Deep-Dream - загрузить заново обученную начальную модель, полученную с помощью трансферного обучения

Я переделал Inception V3 сеть с использованием метода обучения передачи, следуя этой статье.

Для этого я удалил последний сетевой слой и подал в сеть сотни изображений моего лица.

Затем была успешно сгенерирована новая модель: inceptionv3-ft.model


Теперь я хотел бы загрузить эту модель и использовать ее фиксированные веса, чтобы применить мое лицо в качестве "темы" к входному изображению, например google-dream,

Для этого я использую keras Программа, которая загружает модели примерно так:

from keras.applications import inception_v3

# Build the InceptionV3 network with our placeholder.
# The model will be loaded with pre-trained ImageNet weights.
model = inception_v3.InceptionV3(weights='imagenet',
                                 include_top=False)
dream = model.input

Полный код здесь: https://github.com/keras-team/keras/blob/master/examples/deep_dream.py

Итак, как мне загрузить и передать не предварительно обученные, а скорее мои RE-обученные веса модели?

1 ответ

Решение

Просто:

from keras.models import load_model

model = load_model('inceptionv3-ft.model')
dream = model.input
Другие вопросы по тегам