UNPKG

@vladmandic/face-api

Version:

FaceAPI: AI-powered Face Detection & Rotation Tracking, Face Description & Recognition, Age & Gender & Emotion Prediction for Browser and NodeJS using TensorFlow/JS

35 lines (34 loc) 1.06 kB
import * as tf from '../../dist/tfjs.esm'; import { ConvParams } from '../common/index'; import { SeparableConvParams } from '../common/types'; export declare type BatchNorm = { sub: tf.Tensor1D; truediv: tf.Tensor1D; }; export declare type ConvWithBatchNorm = { conv: ConvParams; bn: BatchNorm; }; export declare type MobilenetParams = { conv0: SeparableConvParams | ConvParams; conv1: SeparableConvParams; conv2: SeparableConvParams; conv3: SeparableConvParams; conv4: SeparableConvParams; conv5: SeparableConvParams; conv6?: SeparableConvParams; conv7?: SeparableConvParams; conv8: ConvParams; }; export declare type DefaultTinyYolov2NetParams = { conv0: ConvWithBatchNorm; conv1: ConvWithBatchNorm; conv2: ConvWithBatchNorm; conv3: ConvWithBatchNorm; conv4: ConvWithBatchNorm; conv5: ConvWithBatchNorm; conv6: ConvWithBatchNorm; conv7: ConvWithBatchNorm; conv8: ConvParams; }; export declare type TinyYolov2NetParams = DefaultTinyYolov2NetParams | MobilenetParams;