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时需要用字典定义