fixed testcase + faceLandmarkDetection node example

justadudewhohacks · justadudewhohacks · commit fdc7566cad16 · 2018-11-03T12:39:35.000+01:00
diff --git a/examples-nodejs/faceDetection.ts b/examples-nodejs/faceDetection.ts
@@ -0,0 +1,25 @@
+import '@tensorflow/tfjs-node';
+
+import * as canvas from 'canvas';
+import * as fs from 'fs';
+
+import * as faceapi from '../src';
+
+const { Canvas, Image } = canvas
+faceapi.env.monkeyPatch({ Canvas, Image })
+
+async function run() {
+
+  await faceapi.nets.ssdMobilenetv1.loadFromDisk('../weights')
+
+  const img = await canvas.loadImage('./bbt1.jpg')
+  const detections = await faceapi.detectAllFaces(img)
+
+  const out = faceapi.createCanvasFromMedia(img) as any
+  faceapi.drawDetection(out, detections)
+
+  fs.writeFileSync('./faceDetection.jpg', out.toBuffer('image/jpeg'))
+  console.log(detections)
+}
+
+run()
diff --git a/examples-nodejs/faceLandmarkDetection.ts b/examples-nodejs/faceLandmarkDetection.ts
@@ -0,0 +1,26 @@
+import '@tensorflow/tfjs-node';
+
+import * as canvas from 'canvas';
+import * as fs from 'fs';
+
+import * as faceapi from '../src';
+
+const { Canvas, Image } = canvas
+faceapi.env.monkeyPatch({ Canvas, Image })
+
+async function run() {
+
+  await faceapi.nets.ssdMobilenetv1.loadFromDisk('../weights')
+  await faceapi.nets.faceLandmark68Net.loadFromDisk('../weights')
+
+  const img = await canvas.loadImage('./bbt1.jpg')
+  const results = await faceapi.detectAllFaces(img).withFaceLandmarks()
+
+  const out = faceapi.createCanvasFromMedia(img) as any
+  faceapi.drawDetection(out, results.map(res => res.detection))
+  faceapi.drawLandmarks(out, results.map(res => res.faceLandmarks), { drawLines: true, color: 'red' })
+
+  fs.writeFileSync('./faceLandmarkDetection.jpg', out.toBuffer('image/jpeg'))
+}
+
+run()
diff --git a/examples-nodejs/index.ts b/examples-nodejs/index.ts
diff --git a/test/tests/faceLandmarkNet/FaceLandmark68NetBase.test.ts b/test/tests/faceLandmarkNet/FaceLandmark68NetBase.test.ts
@@ -2,11 +2,28 @@ import * as tf from '@tensorflow/tfjs-core';
 
 import { FaceLandmark68NetBase } from '../../../src/faceLandmarkNet/FaceLandmark68NetBase';
 
+class FakeFaceLandmark68NetBase extends FaceLandmark68NetBase<any> {
+  public runNet(_: any): any {
+  }
+
+  protected getDefaultModelName(): any {
+    throw new Error('FakeNeuralNetwork - getDefaultModelName not implemented')
+  }
+
+  protected extractParams(_: any): any {
+    throw new Error('FakeNeuralNetwork - extractParams not implemented')
+  }
+
+  protected extractParamsFromWeigthMap(_: any): any {
+    throw new Error('FakeNeuralNetwork - extractParamsFromWeigthMap not implemented')
+  }
+}
+
 describe('FaceLandmark68NetBase', () => {
 
   describe('postProcess', () => {
 
-    const net = new FaceLandmark68NetBase('')
+    const net = new FakeFaceLandmark68NetBase('')
 
     describe('single batch', () => {
 
