-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathindex.js
More file actions
45 lines (39 loc) · 1.19 KB
/
index.js
File metadata and controls
45 lines (39 loc) · 1.19 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
import * as tfjs from '@tensorflow/tfjs-node';
import * as faceapi from 'face-api.js';
import * as fs from 'fs';
import * as cv from 'opencv';
const minConfidence = 0.5;
/*
const faceDetectionNet = faceapi.nets.ssdMobilenetv1;
const faceDetectionOptions = new faceapi.SsdMobilenetv1Options({minConfidence});
*/
const faceDetectionNet = faceapi.nets.tinyFaceDetector;
const faceDetectionOptions = new faceapi.TinyFaceDetectorOptions({minConfidence});
let globalFrame = null;
let lastFrame = null;
function processFrame() {
if (globalFrame == null) return;
if (lastFrame == globalFrame) return;
lastFrame = globalFrame;
let tFrame = faceapi.tf.tensor3d(globalFrame, [480, 640, 3])
faceapi.detectAllFaces(tFrame, faceDetectionOptions).then((faces) => {
console.log(faces);
});
}
setInterval(processFrame, 100);
async function run() {
await faceDetectionNet.loadFromDisk('./weights');
let video = '/dev/video0';
let cap = new cv.VideoCapture(video);
let capture = () => {
cap.read(function(err, frame) {
if (frame.width() > 0) {
let data = new Uint8Array(frame.getData().buffer);
globalFrame = data;
}
setTimeout(capture, 0);
});
}
capture();
}
run()