如何在具有N 1类的CNN模型中使用N类加载一个类别的权重文件



我已经建立了一个具有3个类标签的CNN模型,并且为其生成了权重(.h5文件(。我想与同一模型一起使用该权重文件,但唯一的区别是它现在有4个类而不是3个类别。

我尝试使用

checkpoint = model.load_weights("filename.h5") 

,但我总是将错误作为类不匹配。

任何人都可以帮助我提供任何建议

也许在keras中遵循yanboliang的建议:使用model.layers.pop((删除图层不起作用,通过用4个类输出的新层覆盖模型的最后一层?

相关内容

最新更新