如何使用keras / tensorflow将tfjs模型加载到python中

Question

我有一个训练有素的tfjs模型，我想使用keras / tensorflow将其加载到python

我找到了如何将tf和keras model to tfjs models.转换为load tfjs model using js的方法

如何将模型加载到python中？

Answer 1

您可以使用model save API从JS保存tfjs模型。

然后您可以使用tfjs-converter将模型转换为python兼容格式。最后，您应该能够使用keras负载模型API在python中使用load_model。

tf.keras.models.load_model(
   filepath,
   custom_objects=None,
   compile=True
)