采用 setTimeout( () => { }, 0) 能将 CUP计算密集的部分放到 下一轮再运行
setTimeout(() => {
// 模型预测
const pred = tf.tidy(() => {
const x = img2x(img);
return this.model.predict(x);
})
// 将结果转换为 易读数组
const results = pred.arraySync()[0]
.map((score, i) => ({
score, label: this.CLASSES[i]
}))
.sort((a, b) => b.score - a.score);
console.log(results);
}, 0);
网友评论