如何从 tensorflow.js 下载模型和权重

Question

我正在尝试下载一个包含权重的预训练的 tensorflow.js 模型，以在 tensorflow 标准版本的 python 中离线使用，作为一个无论如何都不处于早期阶段的项目的一部分，因此切换到 tensorflow.js 是不是一种可能性。但我无法弄清楚如何下载这些模型，以及是否有必要对模型进行一些转换。

我知道在 javascript 中我可以访问模型并通过像这样调用它们来使用它们，但是如果是这种情况，我如何实际获取 .ckpt 文件或模型？

<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.13.3"></script>

<script src="https://cdn.jsdelivr.net/npm/@tensorflow-models/posenet@0.2.3"></script>

我的最终目标是获得冻结的模型文件，并获得像在 tensorflow 的正常版本中所做的那样的输出。此外，这将用于离线环境，因此任何在线参考都没有用。

感谢您的回复

Answer 1

可以通过调用save模型的方法来保存模型拓扑及其权重。

const model = tf.sequential();
model.add(tf.layers.dense(
     {units: 1, inputShape: [10], activation: 'sigmoid'}));
const saveResult = await model.save('downloads://mymodel');
// This will trigger downloading of two files:
//   'mymodel.json' and 'mymodel.weights.bin'.
console.log(saveResult);

根据保存模型及其权重的位置（localStorage、IndexDB，...），有不同的方案字符串。文档