mirror of https://github.com/vladmandic/human
19 lines
647 B
TypeScript
19 lines
647 B
TypeScript
|
import * as tf from '../../dist/tfjs.esm.js';
|
||
|
|
declare type Tensor = typeof tf.Tensor;
|
||
|
|
declare type DB = Array<{
|
||
|
|
name: string;
|
||
|
|
source: string;
|
||
|
|
embedding: number[];
|
||
|
|
}>;
|
||
|
|
export declare function load(config: any): Promise<any>;
|
||
|
|
export declare function similarity(embedding1: any, embedding2: any, order?: number): number;
|
||
|
|
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;
|
||
|
|
export declare function predict(image: any, config: any): Promise<unknown>;
|
||
|
|
export {};
|