UNPKG

face-api.js

Version:

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

38 lines (37 loc) 1.13 kB
import * as tf from '@tensorflow/tfjs-core'; import { FaceLandmarks5 } from '../classes/FaceLandmarks5'; import { ConvParams, FCParams } from '../common'; import { WithFaceDetection, WithFaceLandmarks } from '../factories'; export declare type SharedParams = { conv1: ConvParams; prelu1_alpha: tf.Tensor1D; conv2: ConvParams; prelu2_alpha: tf.Tensor1D; conv3: ConvParams; prelu3_alpha: tf.Tensor1D; }; export declare type PNetParams = SharedParams & { conv4_1: ConvParams; conv4_2: ConvParams; }; export declare type RNetParams = SharedParams & { fc1: FCParams; prelu4_alpha: tf.Tensor1D; fc2_1: FCParams; fc2_2: FCParams; }; export declare type ONetParams = SharedParams & { conv4: ConvParams; prelu4_alpha: tf.Tensor1D; fc1: FCParams; prelu5_alpha: tf.Tensor1D; fc2_1: FCParams; fc2_2: FCParams; fc2_3: FCParams; }; export declare type NetParams = { pnet: PNetParams; rnet: RNetParams; onet: ONetParams; }; export declare type MtcnnResult = WithFaceLandmarks<WithFaceDetection<{}>, FaceLandmarks5>;