我正在使用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%或更高的结果。
如何降低班级概率阈值?