Загрузка ансамблевой модели keras выдает ValueError: Неверный аргумент input_shape (None, 224, 224, 3): модель имеет 0 тензорных входов - PullRequest
Новая
       80

Загрузка ансамблевой модели keras выдает ValueError: Неверный аргумент input_shape (None, 224, 224, 3): модель имеет 0 тензорных входов

0 голосов
/ 10 ноября 2019

Моя модель представляет собой ансамбль из двух разных моделей керас, модели соединены с одним входным слоем и имеют 2 выходных слоя в сочетании. Обе модели предварительно обучены, и я пытаюсь создать параллельную архитектуру. Моя архитектура: `

model_input = Input(shape=(224,224,3), name="model_input")
gender_model_copy.layers.pop(0)
color_model_copy.layers.pop(0)
color_model_ens1 = color_model_copy(model_input)
gender_model_ens1 = gender_model_copy(model_input)
model_f = Model(input=[model_input], output=[color_model_ens1,gender_model_ens1])
model_f.save('path')

`

Модель компилируется, и я также могу делать прогнозы, но когда я сохраняю ее и пытаюсь перезагрузить, я получаю:

ValueError: Invalid input_shape argument (None, 224, 224, 3): model has 0 tensor inputs. Полная трассировка: Ссылка Gist Gist .

У меня есть пользовательский слой, который я добавляю, используя аргумент custom_objects={'Scale':Scale()} в keras.models.load_model Моя версия keras - 2.2.5 и тензор потока версия 1.15

...