使用keras调用load_model时报错ValueError: Unknown Layer

原因:模型使用了自定的层

处理:在load_model函数中添加custom_objects参数,该参数接受一个字典,键值为自定义的层的函数名(类名)

例如:

自定义了两层

def MaxPoolingWithArgmax2D(layer):

       pass

def MaxUnpooling2D(layer):

       pass

加载模型方式:
model = load_model(args["model"], custom_objects={'MaxPoolingWithArgmax2D': MaxPoolingWithArgmax2D, 'MaxUnpooling2D': MaxUnpooling2D})

你可能感兴趣的:(深度学习)