使用keras调用load_model时报错ValueError: Unknown Layer_valueerror: unknown layer: maxpoolingwithargmax2d

作者：从前慢现在也慢 | 2024-04-28 01:48:50

踩

valueerror: unknown layer: maxpoolingwithargmax2d

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

处理：在load_model函数中添加custom_objects参数，该参数接受一个字典，键值为自定义的层的函数名（类名）

例如：

自定义了两层

def MaxPoolingWithArgmax2D(layer):

pass

def MaxUnpooling2D(layer):

pass

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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/499822