human/types/blazeface/blazeface.d.ts

24 lines
690 B
TypeScript
Raw Normal View History

2021-05-22 20:53:51 +02:00
import { Config } from '../config';
import { Tensor, GraphModel } from '../tfjs/types';
2021-03-17 23:57:00 +01:00
export declare class BlazeFaceModel {
model: GraphModel;
2021-05-22 20:53:51 +02:00
anchorsData: [number, number][];
anchors: Tensor;
2021-03-17 23:57:00 +01:00
inputSize: number;
2021-05-22 20:53:51 +02:00
config: Config;
2021-06-03 15:41:53 +02:00
constructor(model: any, config: Config);
getBoundingBoxes(inputImage: Tensor): Promise<{
2021-03-17 23:57:00 +01:00
boxes: {
2021-05-22 20:53:51 +02:00
box: {
startPoint: Tensor;
endPoint: Tensor;
2021-05-22 20:53:51 +02:00
};
2021-03-17 23:57:00 +01:00
landmarks: any;
2021-04-28 14:58:21 +02:00
anchor: number[];
2021-03-17 23:57:00 +01:00
confidence: number;
}[];
scaleFactor: number[];
} | null>;
}
2021-06-03 15:41:53 +02:00
export declare function load(config: Config): Promise<BlazeFaceModel>;