我正在关注 youtube 上的tensorflowjs 迁移学习课程。我试图让模型识别我上传的一些图片。问题是当我关闭窗口或重新加载时,模型会从头开始加载,并且刚刚学习的数据会丢失。如何在“迁移学习”之后保存模型并在访问该页面时加载它。 (我是第一次从谷歌加载 MobileNet)。谢谢你们!!!
按照 youtube 上的tensorflowjs课程,我可以训练一个模型来识别一些图像。但是当重新加载窗口时,它被处理掉,我必须重新训练。 (我正在制作一个网络来识别用户脸部,每个用户在登录时都可以注册他们的脸部以执行某些任务)
您需要将模型保存在某个地方,现在 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
}
}