ValueError: Unknown layer: KerasLayer

h5模型转换为pb模型的时候,程序中有一条语句是用来载入h5模型的:

model = tf.keras.models.load_model(h5_save_path, compile=False)

结果运行时报错如下:

ValueError: Unknown layer: KerasLayer

解决办法:

model = tf.keras.models.load_model((h5_save_path), custom_objects={'KerasLayer': hub.KerasLayer})

原因分析:搭建网络时,除了最后一层,前面的是使用tensorflow-hub下载下来的,重新load_model时需要用字典定义

你可能感兴趣的:(技术,神经网络,深度学习)