如何使用Keras/Tensorflow将tfjs模型加载到Python中

Question

我有一个预先训练好的tfjs模型，我想使用keras/tensorflow将其加载到Python中。

我找到了如何将tf和keras模型转换为tfjs模型（链接1）。以及如何使用js加载tfjs模型（链接2）。

我该如何将模型加载到Python中？

- thebeancounter

Answer 1

你可以使用模型保存API在JS中保存tfjs模型。

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

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

- BlessedKey

---

我不需要在tf中加载keras模型，我需要在keras中加载tfjs模型。 - thebeancounter

您可以使用模型保存 API 从 JS 中保存 tfjs 模型。接下来，您可以使用转换器将模型转换为 Python 兼容格式。然后，您应该能够在 Python 中使用 load_model。 - BlessedKey

这是问题， 请编辑答案以便我可以接受它。 - thebeancounter

更新完毕。谢谢。希望有所帮助。 - BlessedKey