Как сохранить сглаженный результат CNN? - PullRequest
0 голосов
/ 13 декабря 2018

У меня есть следующая сверточная нейронная сеть для применения к изображениям:

classifier = Sequential()
classifier.add(Convolution2D(128, (3, 3), input_shape = (128, 128, 3), activation = 'relu'))
classifier.add(MaxPooling2D(pool_size = (2, 2)))
classifier.add(Convolution2D(64, (3, 3), activation = 'relu'))
classifier.add(MaxPooling2D(pool_size = (2, 2)))
classifier.add(Flatten())

После применения сверточных слоев и слоев с максимальным объединением я сглаживаю результаты и хочу сохранить только этот результат (позже я хочу работать сэтот результат, используя неконтролируемые методы).Как я могу это сделать?Единственные примеры, которые у меня есть, продолжают процесс подгонки к модели, и я никогда не сохраняю сплющенные слои.

1 Ответ

0 голосов
/ 17 декабря 2018

Это описано в документации Keras для предварительно обученных моделей.См. Примеры об извлечении объектов, https://keras.io/applications/#extract-features-with-vgg16

Когда у вас есть модель, просто выполните:

features = model.predict(x)

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...