node-bebop/examples/opencv-face-tracking.js

Node.js JavaScript library for the Parrot Bebop

"use strict";

//
// This example requires the haarcascade file located at
// https://github.com/peterbraden/node-opencv/raw/master/data/haarcascade_frontalface_alt.xml
//

var bebop = require("../."),
    cv = require("opencv"),
    async = require("async");

var drone = bebop.createClient(),
    mjpg = drone.getMjpegStream(),
    w = new cv.NamedWindow("Video", 0),
    buf = null;

drone.connect(function() {
  drone.MediaStreaming.videoEnable(1);
});

mjpg.on("data", function(data) {
  buf = data;
});

async.forever(
  function(callback) {
    if (buf == null) {
      return callback();
    }

    cv.readImage(buf, function(err, im) {
      if (err) {
        console.log(err);
        callback();
      } else {
        im.detectObject("haarcascade_frontalface_alt.xml", {}, function(e, faces) {
          if (e) {
            console.log(err);
          }

          for (var i = 0; i < faces.length; i++) {
            var face = faces[i];
            im.rectangle([face.x, face.y],
              [face.x + face.width, face.y + face.height], [0, 255, 0], 2);
          }

          w.show(im);
          w.blockingWaitKey(0, 50);

          setTimeout(function() {
            callback();
          }, 100);
        });
      }
    });
  }
);