javascript - JavaScript 中的语义分割

Question

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

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

标签： javascripttensorflowdeep-learningartificial-intelligenceimage-segmentation

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