-
Notifications
You must be signed in to change notification settings - Fork 1
/
index.js
46 lines (36 loc) · 1.19 KB
/
index.js
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
const SSD_MOBILENETV1 = 'ssd_mobilenetv1'
let selectedFaceDetector = SSD_MOBILENETV1
let minConfidence = 0.5
function getFaceDetectorOptions() {
return new faceapi.SsdMobilenetv1Options({ minConfidence })
}
function getCurrentFaceDetectionNet() {
console.log("gcfdn");
console.log(faceapi.nets.ssdMobilenetv1);
return faceapi.nets.ssdMobilenetv1
}
function isFaceDetectionModelLoaded() {
return !!getCurrentFaceDetectionNet().params
}
async function changeFaceDetector(detector) {
console.log("cfd");
if (!isFaceDetectionModelLoaded()) {
console.log("nifdml");
await getCurrentFaceDetectionNet().load('/')
}
}
async function onSelectedFaceDetectorChanged(e) {
selectedFaceDetector = e.target.value
await changeFaceDetector(e.target.value)
updateResults()
}
function initFaceDetectionControls() {
const faceDetectorSelect = $('#selectFaceDetector')
faceDetectorSelect.val(selectedFaceDetector)
faceDetectorSelect.on('change', onSelectedFaceDetectorChanged)
faceDetectorSelect.material_select()
const inputSizeSelect = $('#inputSize')
inputSizeSelect.val(inputSize)
inputSizeSelect.on('change', onInputSizeChanged)
inputSizeSelect.material_select()
}