Keras在原有的模型( VGG16,ResNet 等基础模型)中加入自定义的卷积操作

Value error 或layer 不存在等问题

ValueError: Unknown Layer:LayerName

方法1  (亲测可行!!!) model = load_model(weights_path, by_name=True)  

方法2 (待验证)  在Keras中,如果存在自定义layer或者loss,需要在load_model()中以字典形式指定layer或loss。

model = load_model(weights_path, custom_objects={'my_loss':my_loss,'NestedLSTM': NestedLSTM})

参考:https://blog.csdn.net/liushuijingying2/article/details/87695593

https://blog.csdn.net/larry233/article/details/88569797

你可能感兴趣的:(Keras在原有的模型( VGG16,ResNet 等基础模型)中加入自定义的卷积操作)