如何在ReactJS中保存模型和加载模型

Question

我正在关注 youtube 上的tensorflowjs 迁移学习课程。我试图让模型识别我上传的一些图片。问题是当我关闭窗口或重新加载时，模型会从头开始加载，并且刚刚学习的数据会丢失。如何在“迁移学习”之后保存模型并在访问该页面时加载它。 （我是第一次从谷歌加载 MobileNet）。谢谢你们！！！

按照 youtube 上的tensorflowjs课程，我可以训练一个模型来识别一些图像。但是当重新加载窗口时，它被处理掉，我必须重新训练。 （我正在制作一个网络来识别用户脸部，每个用户在登录时都可以注册他们的脸部以执行某些任务）

Answer 1

您需要将模型保存在某个地方，现在 TensorFlow 可以轻松做到这一点。

为了提高效率，保存模型的好地方是 indexeddb :

await model.save('indexeddb://my-model');

加载应用程序后，您可以检查模型是否已存在。

const loadModel = async (indexddbAdress) => {
    try{
      return await tf.loadGraphModel('indexeddb://' + indexddbAdress);
    } catch (error) {
      // model not in indexddb, maybe start the training process
    }
  }

参考：https://www.tensorflow.org/js/guide/save_load