nodoface/examples/hog_on_video.ts

C++ Addon for fast and efficient face analytics

#!/usr/bin/env ts-node
import * as nodoface from '../';
import { SequenceCapture, FaceDetectorHOG } from '../';

let sequenceCapture = new SequenceCapture();
let HOG = new FaceDetectorHOG();

const argv = process.argv.slice(2);

sequenceCapture.openVideoFile(argv[0]);
HOG.load();
let progress = 0;
for (let i = 0; progress < 1; i++) {
    let img = sequenceCapture.getNextFrame();
    let res = HOG.detectFaces(sequenceCapture.getGrayFrame());
    let bboxes = res.detections;
    progress = sequenceCapture.getProgress();
    console.log(`Frame ${i} detections [Progress ${(100 * progress).toFixed(2)}%]:`);
    
    for (let j = 0; j < bboxes.length; j++) {
        let rect = bboxes[j];
        console.log(`
        Detection ${j}: 
            x:${rect.x.toFixed(0)}, y:${rect.y.toFixed(0)}, 
            width:${rect.width.toFixed(0)}, height:${rect.height.toFixed(0)} 
        `);
        nodoface.drawRect(img, bboxes[j], [0, 255, 0]);
    }
    nodoface.showImage(img);
    nodoface.waitKey(1000 / sequenceCapture.fps);
}
nodoface.destroyAllWindows();