JavaScript 中的语义分割

Question

这是我第二次发布这个问题，但措辞不同。我是人工智能和深度学习的新手。我想制作一个工具来用不同的图案替换图像中的地板。我制作了一个包含近 250 张图片的数据集，并将它们存储在一个 json 文件中。使用 trainMask R-CNN 模型我得到了一些结果（使用 Python），但我想使用 JavaScript。我将我的模型转换为tf.js 格式，但我面临不同的挑战，例如如何在tf.js 中加载模型，以及如何进行预测。

谁能帮助我以一种简单快捷的方式开始对自定义数据集进行分割。我不明白tf.js。

Answer 1

要在 tf.js 中加载模型，您可以使用 tf.loadGraphModel 函数

async function loadModel() {
    model = await tf.loadGraphModel("model/path");
    // do what you need with the model
}

如需预测，请查看tf.js API