Ожидается увидеть 2 массива (ов), но вместо этого получен следующий список из 1 массива - PullRequest
0 голосов
/ 19 марта 2019

Я хочу создать модель, которая получит одно изображение и вычислит изображение двумя Softmax (два вывода).Код:

base_model = InceptionV3(include_top=False)
x = base_model.output
x = GlobalAveragePooling2D()(x)

#  first softmax
x_1 = Dense(1024, activation='relu')(x)
predictions_1 = Dense(4, activation='softmax')(x_1)

#  second Softmax
x_2 = Dense(1024, activation='relu')(x)
predictions_2 = Dense(4, activation='softmax')(x_2)
my_model = Model(inputs=base_model.input, outputs=[predictions_1,predictions_2])

#  train
my_model.compile(...)
my_model.fit_generator(...)

Во время обучения я получил ошибку:

ValueError: Ожидается увидеть 2 массива (ов), но вместо этого получен следующий список из 1 массива:[массив ([[0., 0., 0., 1.], [0., 0., 0., 1.], [0., 0., 0., 1.], [0.,1., 0., 0.], [1., 0., 0., 0.], [0., 1., 0., 0.], [0., 1., 0., 0.], [0., 1., 0., 0.], ...

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