JavaScript中的语义分割

Question

这是我第二次发布此问题，但用不同的词。我是AI和深度学习的新手。我想制作一个工具，以不同的图案替换图像中的地板。我制作了将近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