UNPKG

face-api.js

Version:

JavaScript API for face detection and face recognition in the browser with tensorflow.js

28 lines (27 loc) 1.09 kB
import * as tf from '@tensorflow/tfjs-core'; import { FaceDetection } from '../classes/FaceDetection'; import { NetInput, TNetInput } from '../dom'; import { NeuralNetwork } from '../NeuralNetwork'; import { ISsdMobilenetv1Options } from './SsdMobilenetv1Options'; import { NetParams } from './types'; export declare class SsdMobilenetv1 extends NeuralNetwork<NetParams> { constructor(); forwardInput(input: NetInput): { boxes: tf.Tensor2D[]; scores: tf.Tensor1D[]; }; forward(input: TNetInput): Promise<{ boxes: tf.Tensor2D[]; scores: tf.Tensor1D[]; }>; locateFaces(input: TNetInput, options?: ISsdMobilenetv1Options): Promise<FaceDetection[]>; protected getDefaultModelName(): string; protected extractParamsFromWeigthMap(weightMap: tf.NamedTensorMap): { params: NetParams; paramMappings: import("../common").ParamMapping[]; }; protected extractParams(weights: Float32Array): { params: NetParams; paramMappings: import("../common").ParamMapping[]; }; }