28 lines
1.1 KiB
TypeScript
28 lines
1.1 KiB
TypeScript
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[];
|
|
};
|
|
}
|
|
//# sourceMappingURL=SsdMobilenetv1.d.ts.map
|