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