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

JavaScript 中的语义分割

Semantic Segmentation in JavaScript

javascript

artificial-intelligence

image-segmentation

deep-learning

tensorflow