-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathscript.js
More file actions
72 lines (65 loc) · 2.34 KB
/
script.js
File metadata and controls
72 lines (65 loc) · 2.34 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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
const video = document.getElementById("video");
/* !IMPORTANT!
The name of the folder must be the same as the label
Add your name in the labels array;
TODO: need optimisation. If loaded multiple pictures(models) will crash.
*/
const labels = ["Tiberiu"];
Promise.all([
faceapi.nets.faceRecognitionNet.loadFromUri("/models"),
faceapi.nets.faceLandmark68Net.loadFromUri("/models"),
faceapi.nets.ssdMobilenetv1.loadFromUri("/models")
]).then(startVideo);
async function startVideo() {
Promise.all(
labels.map(async label => {
const descriptions = [];
for (let i = 1; i <= 2; i++) {
const img = await faceapi.fetchImage(
`http://127.0.0.1:5500/labeled_images/${label}/${i}.jpg`
);
const detections = await faceapi
.detectSingleFace(img)
.withFaceLandmarks()
.withFaceDescriptor();
descriptions.push(detections.descriptor);
}
navigator.getUserMedia(
{ video: {} },
stream => (video.srcObject = stream),
err => console.error(err)
);
let data = new faceapi.LabeledFaceDescriptors(label, descriptions);
const labeledFaceDescriptors = data;
const faceMatcher = new faceapi.FaceMatcher(labeledFaceDescriptors, 0.6);
video.addEventListener("play", () => {
const canvas = faceapi.createCanvasFromMedia(video);
document.body.append(canvas);
const displaySize = { width: video.width, height: video.height };
faceapi.matchDimensions(canvas, displaySize);
setInterval(async () => {
canvas.getContext("2d").clearRect(0, 0, canvas.width, canvas.height);
const detections = await faceapi
.detectAllFaces(video)
.withFaceLandmarks()
.withFaceDescriptors();
const resizedDetections = faceapi.resizeResults(
detections,
displaySize
);
const results = resizedDetections.map(d =>
faceMatcher.findBestMatch(d.descriptor)
);
results.forEach((result, i) => {
const box = resizedDetections[i].detection.box;
const drawBox = new faceapi.draw.DrawBox(box, {
label: result.toString()
});
drawBox.draw(canvas);
});
}, 1000);
});
})
);
}
function loadLabeledImages() {}