2021-05-25 14:58:20 +02:00
|
|
|
/**
|
|
|
|
* HSE-FaceRes Module
|
|
|
|
* Returns Age, Gender, Descriptor
|
|
|
|
* Implements Face simmilarity function
|
|
|
|
*/
|
2021-06-03 15:41:53 +02:00
|
|
|
import { Tensor, GraphModel } from '../tfjs/types';
|
|
|
|
import { Config } from '../config';
|
2021-03-21 19:18:51 +01:00
|
|
|
declare type DB = Array<{
|
|
|
|
name: string;
|
|
|
|
source: string;
|
|
|
|
embedding: number[];
|
|
|
|
}>;
|
2021-06-03 15:41:53 +02:00
|
|
|
export declare function load(config: Config): Promise<GraphModel>;
|
|
|
|
export declare function similarity(embedding1: Array<number>, embedding2: Array<number>, order?: number): number;
|
2021-03-21 19:18:51 +01:00
|
|
|
export declare function match(embedding: Array<number>, db: DB, threshold?: number): {
|
|
|
|
similarity: number;
|
|
|
|
name: string;
|
|
|
|
source: string;
|
|
|
|
embedding: number[];
|
|
|
|
};
|
|
|
|
export declare function enhance(input: any): Tensor;
|
2021-06-03 15:41:53 +02:00
|
|
|
export declare function predict(image: Tensor, config: Config, idx: any, count: any): Promise<unknown>;
|
2021-03-21 19:18:51 +01:00
|
|
|
export {};
|