UNPKG

face-api.js

Version:

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

41 lines (40 loc) 1.45 kB
import * as tf from '@tensorflow/tfjs-core'; import { NetInput, TNetInput } from '..'; import { ConvParams, SeparableConvParams } from '../common'; import { NeuralNetwork } from '../NeuralNetwork'; export declare type ConvWithBatchNormParams = BatchNormParams & { filter: tf.Tensor4D; }; export declare type BatchNormParams = { mean: tf.Tensor1D; variance: tf.Tensor1D; scale: tf.Tensor1D; offset: tf.Tensor1D; }; export declare type SeparableConvWithBatchNormParams = { depthwise: ConvWithBatchNormParams; pointwise: ConvWithBatchNormParams; }; export declare type DenseBlock3Params = { conv0: SeparableConvParams | ConvParams; conv1: SeparableConvParams; conv2: SeparableConvParams; }; export declare type DenseBlock4Params = DenseBlock3Params & { conv3: SeparableConvParams; }; export declare type TinyFaceFeatureExtractorParams = { dense0: DenseBlock3Params; dense1: DenseBlock3Params; dense2: DenseBlock3Params; }; export declare type FaceFeatureExtractorParams = { dense0: DenseBlock4Params; dense1: DenseBlock4Params; dense2: DenseBlock4Params; dense3: DenseBlock4Params; }; export interface IFaceFeatureExtractor<TNetParams extends TinyFaceFeatureExtractorParams | FaceFeatureExtractorParams> extends NeuralNetwork<TNetParams> { forwardInput(input: NetInput): tf.Tensor4D; forward(input: TNetInput): Promise<tf.Tensor4D>; }