let pose; // to identify the keypoints of the body

let skeleton;  // for drawing the line between the points

​

​

let min_confidence = .2;

let raw_ht_pos;

let smooth_ht_pos;

let raw_body_dist = 1.;

let smooth_body_dist = 1.;

let smooth_factor = .2;

​

​

// -----------------------------------------------------------

// -----------------------------------------------------------

// -----------------------------------------------------------

​

function run_detection(){

​

  // video.size(640,480);

  raw_ht_pos = createVector(0,0);

  smooth_ht_pos = createVector(0,0);

​

  posenet = ml5.poseNet(video, () => {

          console.log('model loaded');

     background(255);

      });

​

​

      posenet.on('pose', (poses) => {

​

          if (poses.length > 0) {

              pose = poses[0].pose;

             // skeleton = poses[0].skeleton;

          }

   });

  // Hide the video element, and just show the canvas

  video.hide();

​

​

​

}

​

// -----------------------------------------------------------

// -----------------------------------------------------------

// -----------------------------------------------------------

​

// A function to draw ellipses over the detected keypoints

function drawKeypoints(pose)  {

​

    // For each pose detected, loop through all the keypoints

    //let pose = poses[0].pose;

    for (let j = 0; j < pose.keypoints.length; j++) {

      // A keypoint is an object describing a body part (like rightArm or leftShoulder)

      let keypoint = pose.keypoints[j];

      // Only draw an ellipse is the pose probability is bigger than 0.2

      if (keypoint.score > min_confidence) {

        fill(255);

        noStroke();

         ellipse(keypoint.position.x, keypoint.position.y, 11, 11);

      }

    }

}

​

 // -----------------------------------------------------------

// -----------------------------------------------------------

// -----------------------------------------------------------

​

// A function to draw the skeletons

function drawSkeleton(skeleton) {

    for (let j = 0; j < skeleton.length; j++) {

      let partA = skeleton[j][0];

      let partB = skeleton[j][1];

      strokeWeight(3);

      stroke(155);

      line(partA.position.x, partA.position.y, partB.position.x, partB.position.y);

    }

 // }

}

​

​

function modelReady() {

  console.log('Model Loaded');

 // 

}

​

​

​

// -----------------------------------------------------------

// -----------------------------------------------------------

// -----------------------------------------------------------

​

function advancedHeadtracking(pose){

​

  // take nose ant both eyes an get center pos of them

  let nosepos = pose.keypoints[0].position;

  let eye1pos = pose.keypoints[1].position;

  let eye2pos = pose.keypoints[2].position;

  let _x = (nosepos.x + eye1pos.x +  eye2pos.x)*.333333;

  let _y = (nosepos.y + eye1pos.y +  eye2pos.y)*.333333;

  raw_ht_pos.x = _x;

  raw_ht_pos.y = _y;

  smooth_ht_pos.x = lerp(smooth_ht_pos.x,raw_ht_pos.x,smooth_factor);

  smooth_ht_pos.y = lerp(smooth_ht_pos.y,raw_ht_pos.y,smooth_factor);

  // both distance from nose to eye

  let _d1 = dist(nosepos.x,nosepos.y,eye1pos.x,eye1pos.y);

  let _d2 = dist(nosepos.x,nosepos.y,eye2pos.x,eye2pos.y);

  let _d3 = dist(eye1pos.x,eye1pos.y,eye2pos.x,eye2pos.y);

  raw_body_dist = (_d1+_d2+_d3)/3;

  smooth_body_dist = lerp(smooth_body_dist ,raw_body_dist,smooth_factor);

​

}

​

​

​