python - 如何在两个图像类别（二进制）上重用具有多个图像类别的预训练模型

Question

我已经在 keras 中训练了我的模型，将食物图像分类为 16 个类别。我每个类别有 1000 个样本。在这种情况下，我的模型做得很好，准确率约为 95%。现在我想在同一组图像上重用这个模型，但现在分为两类（健康和不健康）。

我试图这样做：

model = load_model('model_saved.h5')
ll = model.layers[len(model.layers)-2].output
ll = Dense(1,activation="sigmoid",name="densehh_out")(ll)

model = Model(inputs=model.input,outputs=ll)

但准确率从大约 50% 开始，不会超过 72%/

loss 对我的 val_data 也很敏感。

有没有更好的方法来实现更好的精度和收敛损失函数？

标签： pythonmachine-learningkeras