//

// This software is released under the MIT License.

// https://opensource.org/licenses/MIT

​

/* ===

ml5 Example

BodyPix

=== */

​

let bodypix;

let video;

let segmentation;

​

let options = {

  outputStride: 16, //adjust the output stride and see which one works best!

  multiSegmentation: false,

  segmentBodyParts: true,

  flipHorizontal: true,

};

​

function setup() {

  background(0);

  createCanvas(480, 360);

  // Create the video

  video = createCapture(VIDEO);

  video.size(width, height);

  video.hide();

  bodypix = ml5.bodyPix(video, options, modelReady);

  bodypix.on("bodypix", gotResults);

}

​

// Event for body segmentation

function gotResults(result) {

      background(0);

​

  // Save the latest part mask from the model in global variable "segmentation"

  segmentation = result;

  //Draw the video

  //image(video, 0, 0, width, height);

  image(segmentation.partMask, 0, 0, width, height);

  //tint(255, 128); //opacity tuning

}

​

// Event when model is loaded

function modelReady() {

  console.log("Model ready!");

}

​

function draw() {

}

​