javascript - 如何使用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
)