Viene de la sección anterior.
Modificar la parte del código sketch.js
let capture;
let classifier;
let flag = 0;
function ModelReady() {
// 效果,加载图片之后缓慢输出model is ready
console.log('model is ready');
flag2 = 1;
classifier.predict(capture, GetResult);
}
function GetResult(error, results){
if(error){
console.error(error);
}else{
console.log(results);
// 第一个预测值
let name = results[0].className;
let prob = results[0].probability;
// 设置颜色
fill([0,255,0]);
textSize(25);
image(capture, 0, 0, width, height);
// 将预测结果输出
text(name, 10, height-20);
flag = 1;
}
}
function setup(){
createCanvas(480, 480);
capture = createCapture(VIDEO); //camera capture function
capture.hide();
classifier = ml5.imageClassifier('MobileNet', capture, ModelReady);
}
function draw(){
if(flag == 1){
classifier.predict(capture, GetResult);
}
}
Después de ejecutar, encienda la cámara de la computadora para realizar el reconocimiento de objetos