xxxxxxxxxx
79
// Classifier Variable
let classifier;
//Model URL
let imageModelURL = 'https://teachablemachine.withgoogle.com/models/y51s-2M-i/';
//video
let video;
let flippedvideo
//store classification
let label= "";
let conf;
let vid1
let vid2
//load the model
function preload() {
classifier = ml5.imageClassifier(imageModelURL + 'model.json');
}
function setup() {
createCanvas(400, 400);
//create the videos
vid1 = createVideo("vid1.mp4")
vid1.hide()
vid2 = createVideo("vid2.mp4")
vid2.hide()
video = createCapture(VIDEO);
video.size(video.width, video.height);
video.hide();
flippedVideo = video
// Start classifying
classifyVideo();
}
function classifyVideo() {
//flippedVideo = ml5.flipImage(video)
classifier.classify(flippedVideo, gotResult);
//flippedVideo.remove();
}
function gotResult(error, results) {
// If there is an error
if (error) {
console.error(error);
return;
}
label = results[0].label;
conf = results[0].confidence
// console.log(conf)
// console.log(results[0]);
// Classify again!
classifyVideo();
}
function draw() {
background(200)
// image(video,0,0)
showvideo()
}
function showvideo(){
if (conf > 0.7) {
if (label == "happy") {
image(vid1, 0, 0)
}
if (label == "neutral") {
image(vid2, 0, 0)
}
}
}