-
Notifications
You must be signed in to change notification settings - Fork 10
/
Copy pathfacialexpression.js
46 lines (37 loc) · 1.6 KB
/
facialexpression.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
const imageUpload = document.getElementById('imageUpload')
Promise.all([
faceapi.nets.tinyFaceDetector.loadFromUri('/models'),
faceapi.nets.faceRecognitionNet.loadFromUri('/models'),
faceapi.nets.faceLandmark68Net.loadFromUri('/models'),
faceapi.nets.ssdMobilenetv1.loadFromUri('/models'),
faceapi.nets.faceExpressionNet.loadFromUri('/models')
]).then(start)
async function start() {
const container = document.createElement('div')
container.style.position = 'relative'
container.width=400
container.height=350
container.style.marginTop="50px"
document.getElementById('uploads').append(container)
let image
let canvas
imageUpload.addEventListener('change', async () => {
if (image) image.remove()
if (canvas) canvas.remove()
image = await faceapi.bufferToImage(imageUpload.files[0])
image.width=400
image.height=350
container.append(image)
canvas = faceapi.createCanvasFromMedia(image)
container.append(canvas)
document.getElementById('load').innerHTML='Predecting...'
const displaySize = { width: image.width, height: image.height }
faceapi.matchDimensions(canvas, displaySize)
const detections = await faceapi.detectAllFaces(image, new faceapi.TinyFaceDetectorOptions()).withFaceLandmarks().withFaceExpressions()
const resizedDetections = faceapi.resizeResults(detections, displaySize)
canvas.getContext('2d').clearRect(0, 0, canvas.width, canvas.height)
faceapi.draw.drawDetections(canvas, resizedDetections)
faceapi.draw.drawFaceExpressions(canvas, resizedDetections)
document.getElementById('load').innerHTML=''
})
}