Question

这是我的模特

    model = VGG16(include_top=False, input_shape=(224, 224, 3))
    x = Flatten()(model.outputs)
    x = Dense(512)(x)
    x = Activation('relu')(x)
    x = Dropout(0.5)(x)
    output1 = Dense(1, activation = 'sigmoid')(x)
    output2 = Dense(1, activation = 'sigmoid')(x)
    output3 = Dense(1, activation = 'sigmoid')(x)
    output4 = Dense(1, activation = 'sigmoid')(x)
    output5 = Dense(1, activation = 'sigmoid')(x)
    output6 = Dense(1, activation = 'sigmoid')(x)
    output7 = Dense(1, activation = 'sigmoid')(x)
    output8 = Dense(1, activation = 'sigmoid')(x)
    model = Model(model.inputs,[output1,output2,output3,output4,output5, output6, output7, output8])
    print(model.summary())
    model.compile(optimizers.rmsprop(lr = 0.0001, decay = 1e-6),
    loss = ["binary_crossentropy","binary_crossentropy","binary_crossentropy","binary_crossentropy", "binary_crossentropy","binary_crossentropy","binary_crossentropy","binary_crossentropy"],metrics = ["accuracy"])

但是当我运行它时，我会得到

InvalidArgumentError: Matrix size-incompatible: In[0]: [1,2508800], In[1]: [25088,512]
     [[{{node dense_45/MatMul}}]]
     [[{{node loss_3/dense_49_loss/Mean_3}}]]

任何解决此问题的建议都会非常有帮助，谢谢。

Answer 1

已通过修复此问题

model = VGG16(include_top=False, input_shape=(224, 224, 3))
x = model.output
x = Flatten()(x)
x = Dense(512)(x)
x = Activation('relu')(x)
x = Dropout(0.5)(x)
output1 = Dense(1, activation = 'sigmoid')(x)
output2 = Dense(1, activation = 'sigmoid')(x)
output3 = Dense(1, activation = 'sigmoid')(x)
output4 = Dense(1, activation = 'sigmoid')(x)
output5 = Dense(1, activation = 'sigmoid')(x)
output6 = Dense(1, activation = 'sigmoid')(x)
output7 = Dense(1, activation = 'sigmoid')(x)
output8 = Dense(1, activation = 'sigmoid')(x)
model = Model(model.inputs,[output1,output2,output3,output4,output5, output6, output7, output8])
print(model.summary())
model.compile(optimizers.rmsprop(lr = 0.0001, decay = 1e-6),
loss = ["binary_crossentropy","binary_crossentropy","binary_crossentropy","binary_crossentropy", "binary_crossentropy","binary_crossentropy","binary_crossentropy","binary_crossentropy"],metrics = ["accuracy"])

矩阵大小不兼容的keras

1 个答案: