Как сохранить разные модели керас в один .h5py - PullRequest
0 голосов
/ 01 октября 2019

Я пытаюсь выполнить классификацию видео с использованием Conv2D и LSTM. После получения функций от Conv2D я передаю их в LSTM. Поскольку это две разные модели, как я могу объединить их в одну для получения одного .h5py?

А также функции из Conv2D передаются в LSTM в виде последовательностей кадров, сохраненных в виде файлов массива ".npy".

Мне нужно сохранить модель для разных целей.

1 Ответ

0 голосов
/ 01 октября 2019

Предполагая, что вы используете tf keras для слияния, вы можете использовать этот метод:

model = keras.models.Model(inputs=[Conv2D, LSTM], outputs=out)

И для сохранения вашей модели вы можете просто использовать .save, как описано здесь

model.save('your_model.h5')

Не совсем уверен, что это то, что вам нужно, но надеюсь, что это поможет.

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