如何使用coco-ssd设置类别概率阈值?

时间:2020-03-13 18:40:29

标签: node.js tensorflow

我正在使用tfjs-node,正在使用@tensorflow-models/coco-ssd

const tf = require('@tensorflow/tfjs');
const tfnode = require('@tensorflow/tfjs-node');
const cocoSsd = require("@tensorflow-models/coco-ssd");
...
const model = await cocoSsd.load();
const predictions = await model.detect(image);

model.detect函数对我有用,但是我希望能够以较低的分类概率查看结果。目前,model.detect仅返回类别概率为50%或更高的结果。

如何降低班级概率阈值?

0 个答案:

没有答案