pmr-core/src/face/recognition/recogniser-base.ts

Core library of Poor Man's Rekognition project under CCExtractor Development

import { Tensor3D } from '@tensorflow/tfjs-node';
import { FaceBlob } from '../face-blob';

export interface FaceRecogConfig {
    recogniser?: 'facenet' // | 'arcface'; // though arcface not working yet
    distanceThreshold?: number; // 1.15
    knownFaces?: FaceBlob[];
}

export interface FaceRecog {

    embedding(faceImage: Tensor3D): Promise<Float32Array>;

    distance(embedding1: Float32Array, embedding2: Float32Array): number;
}