oveddan · May 25, 2018 18:58
diff --git a/index.html b/index.html
 <html>
  <head>
    <script src="https://cdnjs.cloudflare.com/ajax/libs/p5.js/0.6.1/p5.js"></script>
    <script src="https://cdnjs.cloudflare.com/ajax/libs/p5.js/0.6.1/addons/p5.dom.min.js"></script>
    <!-- Load TensorFlow.js -->
    <script src="https://unpkg.com/@tensorflow/tfjs"></script>
    <!-- Load Posenet -->
    <script src="https://unpkg.com/@tensorflow-models/posenet"></script>
    <script src="sketch.js"></script>
  </head>
  <body>
  </body>
 </html>
diff --git a/sketch.js b/sketch.js
 var width = 600;
 var height = 600;
 var videoSize = 600;

 // the size of the neural network model to load. Must be 0.50, 0.75, 1.00, or 1.01
 // The higher the number, the larger the model and the more accurate it is, but
 // the slower the speed.
 var modelSize = 0.75;

 // A number between 0.2 and 1.0. How much posenet should scale the image by before feeding
 // it through the network.  Set this number lower to scale down the image and increase
 // the speed at the cost of accuracy.
 var imageScaleFactor = 0.75;

 // the minimum score of keypoints from posenet to show.
 // Should be between 0.0 and 1.0. Use this to filter out
 // lower accuracy parts
 var minPartConfidence = 0.3;

 // if the pose results should be flipped horizontally. Useful for webcam videos.
 var flipHorizontal = false;

 var capture;
 var net;

 var keypoints = [];

 function estimatePoses() {
  // call posenet to estimate a pose
  net.estimateSinglePose(capture.elt, imageScaleFactor, flipHorizontal)
    .then(function(pose) {
      // store the keypoints from the pose to draw it below
      keypoints = pose.keypoints;
      // next animation loop, call posenet again to estimate poses
      requestAnimationFrame(function() {
        estimatePoses();
      });
    });
 }

 function setup() {
  createCanvas(600, 600);
  // create video capture.  For PoseNet, videos must be square
  capture = createCapture({
    video: {
      width: videoSize,
      height: videoSize
    }
  });
  capture.size(videoSize, videoSize);
  capture.hide();

  // load posenet by downloading the weights for the model.
  posenet.load(modelSize).then(function(loadedNet) {
    net = loadedNet;
    // when it's loaded, start estimating poses
    requestAnimationFrame(function() {
      estimatePoses();
    });
  })
 }

 function draw() {
  background(255);
  image(capture, 0, 0, videoSize, videoSize);

  noStroke();
  // draw keypoints
  for(var i = 0; i < keypoints.length; i++) {
    var keypoint = keypoints[i];
    // filter out keypoints that have a low confidence
    if (keypoint.score > minPartConfidence) {
      // for wrists, make the part red
      if (i == posenet.partIds['leftWrist'] || i == posenet.partIds['rightWrist'])
        fill(255, 0, 0);
      // all other parts are yello
      else
        fill(255, 255, 0);

      ellipse(keypoint.position.x, keypoint.position.y, 10, 10,);
    }
  }

  // get skeleton, filtering out parts wtihout
  // a high enough confidence level
  if (keypoints.length > 0) {
    stroke(255, 255, 0);
    var skeleton = posenet.getAdjacentKeyPoints(keypoints, minPartConfidence);
    for(var i = 0; i < skeleton.length; i++) {
      // draw each line in the skeleton
      var segment = skeleton[i];
      line(
        segment[0].position.x, segment[0].position.y,
        segment[1].position.x, segment[1].position.y
      );
    }
  }
 }
	<html>
	<head>
	<script src="https://cdnjs.cloudflare.com/ajax/libs/p5.js/0.6.1/p5.js"></script>
	<script src="https://cdnjs.cloudflare.com/ajax/libs/p5.js/0.6.1/addons/p5.dom.min.js"></script>
	<!-- Load TensorFlow.js -->
	<script src="https://unpkg.com/@tensorflow/tfjs"></script>
	<!-- Load Posenet -->
	<script src="https://unpkg.com/@tensorflow-models/posenet"></script>
	<script src="sketch.js"></script>
	</head>
	<body>
	</body>
	</html>
	var width = 600;
	var height = 600;
	var videoSize = 600;

	// the size of the neural network model to load. Must be 0.50, 0.75, 1.00, or 1.01
	// The higher the number, the larger the model and the more accurate it is, but
	// the slower the speed.
	var modelSize = 0.75;

	// A number between 0.2 and 1.0. How much posenet should scale the image by before feeding
	// it through the network. Set this number lower to scale down the image and increase
	// the speed at the cost of accuracy.
	var imageScaleFactor = 0.75;

	// the minimum score of keypoints from posenet to show.
	// Should be between 0.0 and 1.0. Use this to filter out
	// lower accuracy parts
	var minPartConfidence = 0.3;

	// if the pose results should be flipped horizontally. Useful for webcam videos.
	var flipHorizontal = false;

	var capture;
	var net;

	var keypoints = [];

	function estimatePoses() {
	// call posenet to estimate a pose
	net.estimateSinglePose(capture.elt, imageScaleFactor, flipHorizontal)
	.then(function(pose) {
	// store the keypoints from the pose to draw it below
	keypoints = pose.keypoints;
	// next animation loop, call posenet again to estimate poses
	requestAnimationFrame(function() {
	estimatePoses();
	});
	});
	}

	function setup() {
	createCanvas(600, 600);
	// create video capture. For PoseNet, videos must be square
	capture = createCapture({
	video: {
	width: videoSize,
	height: videoSize
	}
	});
	capture.size(videoSize, videoSize);
	capture.hide();

	// load posenet by downloading the weights for the model.
	posenet.load(modelSize).then(function(loadedNet) {
	net = loadedNet;
	// when it's loaded, start estimating poses
	requestAnimationFrame(function() {
	estimatePoses();
	});
	})
	}

	function draw() {
	background(255);
	image(capture, 0, 0, videoSize, videoSize);

	noStroke();
	// draw keypoints
	for(var i = 0; i < keypoints.length; i++) {
	var keypoint = keypoints[i];
	// filter out keypoints that have a low confidence
	if (keypoint.score > minPartConfidence) {
	// for wrists, make the part red
	if (i == posenet.partIds['leftWrist'] \|\| i == posenet.partIds['rightWrist'])
	fill(255, 0, 0);
	// all other parts are yello
	else
	fill(255, 255, 0);

	ellipse(keypoint.position.x, keypoint.position.y, 10, 10,);
	}
	}

	// get skeleton, filtering out parts wtihout
	// a high enough confidence level
	if (keypoints.length > 0) {
	stroke(255, 255, 0);
	var skeleton = posenet.getAdjacentKeyPoints(keypoints, minPartConfidence);
	for(var i = 0; i < skeleton.length; i++) {
	// draw each line in the skeleton
	var segment = skeleton[i];
	line(
	segment[0].position.x, segment[0].position.y,
	segment[1].position.x, segment[1].position.y
	);
	}
	}
	}