ValueError при использовании полностью подключенного слоя - PullRequest
0 голосов
/ 18 февраля 2019

В качестве последнего слоя я получил следующий плотный слой.

flat = Flatten()(reduction_pose)

pose = Dense(1024, activation='relu', name='pose')(flat)

xyz = Dense(2, name='xyz')(pose)

model_out = Model(input=input, output=[xyz])

Я бы хотел заменить плотный слой полностью связанным слоем.как я могу это сделать?Я пробовал следующее,

conv10 = Convolution2D(2, (1, 1), name='conv_10_1')(reduction_pose)
x = Reshape((224 * 224, 2))(conv10)
x = Activation('softmax')(x)
outputs = Reshape((224, 224, 2))(x)

model_out = Model(inputs=input, outputs=outputs)

Но я получаю следующую ошибку,

ValueError: общий размер нового массива должен быть неизменным

Ошибка указывает на x = Reshape((224 * 224, 2))(conv10)

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