Я хочу создать модель, которая получит одно изображение и вычислит изображение двумя 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.], ...