// The Coding Train / Daniel Shiffman

// https://thecodingtrain.com/Courses/ml5-beginners-guide/7.1-posenet.html

// https://youtu.be/OIo-DIOkNVg

// https://editor.p5js.org/codingtrain/sketches/ULA97pJXR

​

let video;

let poseNet;

let pose;

let skeleton;

let checked =1;

​

function setup() {

  createCanvas(640, 480);

  video = createCapture(VIDEO);

  video.hide();

  poseNet = ml5.poseNet(video, modelLoaded);

  poseNet.on('pose', gotPoses);

}

​

function gotPoses(poses) {

  if(checked == 1) {

    console.log(poses);

    checked = 0;

  }

  if (poses.length > 0) {

    pose = poses[0].pose;

    skeleton = poses[0].skeleton;

  }

}

​

​

function modelLoaded() {

  console.log('poseNet ready');

}

​

function draw() {

image(video, 0, 0)

​

  if (pose) {

    fill(255, 0, 0);

    ellipse(pose.nose.x, pose.nose.y, 20);

  }

    ;

}