这是我第二次发布这个问题,但用的是不同的措辞。我是AI和深度学习的新手。我想制作一个工具,用不同的图案替换图像中的地板。我制作了一个包含大约250张图片的数据集,并将它们存储在一个json文件中。使用trainMask R-CNN模型,我在Python中获得了一些结果,但我想使用JavaScript。我将我的模型转换成了tf.js
格式,但我遇到了不同的挑战,比如如何在tf.js
中加载模型,以及如何进行预测。
有谁能帮助我以一种简单快捷的方式开始对自定义数据集进行分割?我不太理解tf.js
。
回答:
要在tf.js中加载模型,你可以使用tf.loadGraphModel函数
async function loadModel() { model = await tf.loadGraphModel("model/path"); // 使用模型做你需要做的事情}
关于预测,请查看tf.js API