age gender recognition nodejs example

justadudewhohacks · justadudewhohacks · commit 84ec1267aaf1 · 2019-05-07T21:12:07.000+02:00
diff --git a/examples/examples-nodejs/ageAndGenderRecognition.ts b/examples/examples-nodejs/ageAndGenderRecognition.ts
@@ -0,0 +1,33 @@
+import * as faceapi from 'face-api.js';
+
+import { canvas, faceDetectionNet, faceDetectionOptions, saveFile } from './commons';
+
+async function run() {
+
+  await faceDetectionNet.loadFromDisk('../../weights')
+  await faceapi.nets.faceLandmark68Net.loadFromDisk('../../weights')
+  await faceapi.nets.ageGenderNet.loadFromDisk('../../weights')
+
+  const img = await canvas.loadImage('../images/bbt1.jpg')
+  const results = await faceapi.detectAllFaces(img, faceDetectionOptions)
+    .withFaceLandmarks()
+    .withAgeAndGender()
+
+  const out = faceapi.createCanvasFromMedia(img) as any
+  faceapi.draw.drawDetections(out, results.map(res => res.detection))
+  results.forEach(result => {
+    const { age, gender, genderProbability } = result
+    new faceapi.draw.DrawTextField(
+      [
+        `${faceapi.round(age, 0)} years`,
+        `${gender} (${faceapi.round(genderProbability)})`
+      ],
+      result.detection.box.bottomLeft
+    ).draw(out)
+  })
+
+  saveFile('ageAndGenderRecognition.jpg', out.toBuffer('image/jpeg'))
+  console.log('done, saved results to out/ageAndGenderRecognition.jpg')
+}
+
+run()
diff --git a/examples/examples-nodejs/faceExpressionRecognition.ts b/examples/examples-nodejs/faceExpressionRecognition.ts
@@ -5,10 +5,12 @@ import { canvas, faceDetectionNet, faceDetectionOptions, saveFile } from './comm
 async function run() {
 
   await faceDetectionNet.loadFromDisk('../../weights')
+  await faceapi.nets.faceLandmark68Net.loadFromDisk('../../weights')
   await faceapi.nets.faceExpressionNet.loadFromDisk('../../weights')
 
   const img = await canvas.loadImage('../images/surprised.jpg')
   const results = await faceapi.detectAllFaces(img, faceDetectionOptions)
+  .withFaceLandmarks()
     .withFaceExpressions()
 
   const out = faceapi.createCanvasFromMedia(img) as any
