Building a Face Recognition based Device Control Block

<div>Face Recognition based Device Control Block</div>

<button type="button" onclick="init()">Start</button>

<div id="webcam-container"></div>

<div id="label-container"></div>

<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest/dist/tf.min.js"></script>

<script src="https://cdn.jsdelivr.net/npm/@teachablemachine/image@latest/dist/teachablemachine-image.min.js"></script>

<script src="https://code.jquery.com/jquery-1.9.1.js"></script>

<script type="text/javascript">

    // More API functions here:

    // https://github.com/googlecreativelab/teachablemachine-community/tree/master/libraries/image

    // The link to your model provided by Teachable Machine export panel

    const URL = "https://teachablemachine.withgoogle.com/models/0A687Lx2F/";

    var found1 = false;

    var found2 = false;

    let model, webcam, labelContainer, maxPredictions;

    // Load the image model and setup the webcam

    async function init() {

        const modelURL = URL + "model.json";

        const metadataURL = URL + "metadata.json";

        // load the model and metadata

        // Refer to tmImage.loadFromFiles() in the API to support files from a file picker

        // or files from your local hard drive

        // Note: the pose library adds "tmImage" object to your window (window.tmImage)

        model = await tmImage.load(modelURL, metadataURL);

        maxPredictions = model.getTotalClasses();

        // Convenience function to setup a webcam

        const flip = true; // whether to flip the webcam

        webcam = new tmImage.Webcam(200, 200, flip); // width, height, flip

        await webcam.setup(); // request access to the webcam

        await webcam.play();

        window.requestAnimationFrame(loop);

        // append elements to the DOM

        document.getElementById("webcam-container").appendChild(webcam.canvas);

        labelContainer = document.getElementById("label-container");

        for (let i = 0; i < maxPredictions; i++) { // and class labels

            labelContainer.appendChild(document.createElement("div"));

        }

    }

    async function loop() {

        webcam.update(); // update the webcam frame

        await predict();

        window.requestAnimationFrame(loop);

    }

    // run the webcam image through the image model

    async function predict() {

        // predict can take in an image, video or canvas html element

        const prediction = await model.predict(webcam.canvas);

        for (let i = 0; i < maxPredictions; i++) {

            var classPrediction = prediction[i].className;

            console.log("Predicted: ");

            console.log(classPrediction);

            console.log(classPrediction.localeCompare("Present"));

            if (((classPrediction.localeCompare("Present") == 0) && (prediction[i].probability.toFixed(2)*100) > 80)) {

                console.log("Present > 80 %");

                if (found1 == false) {

                   console.log("posting turn on");

                   $.post('https://api.thingspeak.com/talkbacks/99999/commands.json?api_key=XXXXXXXXXXXXXXXX&command_string=TURN_ON&position=1'); 

                   found1 = true;  // if found is 'true', don't send TALKBACK again

                   found2= false;

                   console.log("found 1 is true");

                   return;

                }

            } 

            else if (((classPrediction.localeCompare("Absent") == 0) && (prediction[i].probability.toFixed(2)*100) > 80)) {

                console.log("Absent > 80 %");

                if (found2 == false) {

                   console.log("posting turn off");

                   $.post('https://api.thingspeak.com/talkbacks/99999/commands.json?api_key=XXXXXXXXXXXXXXXX&command_string=TURN_OFF&position=1'); 

                   found2 = true;  // if found is 'true', don't send TALKBACK again

                   found1 = false;

                   console.log("found 1 is false");

                   return;

                }

            }  

            classPrediction = prediction[i].className + ": " + prediction[i].probability.toFixed(2);

            labelContainer.childNodes[i].innerHTML = classPrediction; 

        }

    }

</script>