123456789101112131415161718192021222324252627 |
- 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[];
- };
- }
|