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
TypeScript
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>;
}