watsonb8/homebridge-face-location
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Successfully training models based on a collection of images

Browse Source
This commit is contained in:
watsonb8
2020-11-08 20:57:57 -05:00
parent e1902a362e
commit fda68e7144
7 changed files with 176 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/index.ts
View File

@ -1,26 +1,11 @@
import { Rtsp } from "rtsp-stream/lib";
// import nodejs bindings to native tensorflow,
// not required, but will speed up things drastically (python required)
import * as faceapi from "face-api.js";
// implements nodejs wrappers for HTMLCanvasElement, HTMLImageElement, ImageData
const canvas = require("canvas");
// patch nodejs environment, we need to provide an implementation of
// HTMLCanvasElement and HTMLImageElement
const { Canvas, Image, ImageData } = canvas;
faceapi.env.monkeyPatch({ Canvas, Image, ImageData });
const main = async () => {
const rtsp = new Rtsp("rtsp://brandon:asdf1234@192.168.1.229/live", {
rate: 10,
});
await faceapi.nets.ssdMobilenetv1.loadFromDisk("./resources");
rtsp.on("data", async (data: Buffer) => {
const input = await canvas.loadImage(data);
const detections = await faceapi.detectAllFaces(input);
console.log();
process.exit(0);
});