主要介绍了keras实现调用自己训练的模型,并去掉全连接层，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

网上的教程大多数是教大家如何加载自定义模型和函数，如下图 这个SelfAttention层是在训练过程自己定义的一个class，但如果要加载这个自定义层，需要在load_model里添加custom_objects字典，这个自定义的类，不要用import ，最好是直接复制进再训练的模型中，这些是基本教程。 ——————分割线讲重点—————— 如果直接运行上面的代码，会出现一个init初始化错误，如下图， 再来看看 这个SelfAttention 自定义的类的初始化 这就说明再调用这个类的

其实很简单 from keras.models import load_model base_model = load_model('model_resenet.h5')#加载指定的模型 print(base_model.summary())#输出网络的结构图 这是我的网络模型的输出，其实就是它的结构图 ______________________________________________________________________________________________