Как использовать predict в предобученной сети keras tf?

Question

[AlexBoss]

День добрый, как использовать predict в предобученной сети keras tf?

Я дообучил готовую модель, путем добавления

model = Sequential()

model.add(Dense(256, activation='relu', input_dim=5 * 5 * 2048))
model.add(Dropout(0.5))
model.add(Dense(36, activation='softmax'))

Само собой, ловлю ошибку

ValueError: Input 0 of layer sequential_3 is incompatible with the layer: expected axis -1 of input shape to have value 51200 but received input with shape [None, 150, 150, 3]

Как можно это обойти?

Comments

[AlexBoss]

Сделал только вариант с загрузкой обеих сетей, сначала прогон через предобученную, потом через свою, но решение не особо нравится

[Сергей Соколов]

почему на вход идёт картинка 150*150*3 ( = 67500) , когда ожидается плоский 5 * 5 * 2048 = 51200

[AlexBoss]

Сергей Соколов, я об этом и говорю, как мне картинку 150*150*3 преобразовать в слой 5 * 5 * 2048