12345678910111213141516171819202122232425262728 |
- import * as tf from '@tensorflow/tfjs-core';
- import { NetInput, TNetInput } from '../dom';
- import { NeuralNetwork } from '../NeuralNetwork';
- import { IMtcnnOptions } from './MtcnnOptions';
- import { MtcnnResult, NetParams } from './types';
- export declare class Mtcnn extends NeuralNetwork<NetParams> {
- constructor();
- load(weightsOrUrl: Float32Array | string | undefined): Promise<void>;
- loadFromDisk(filePath: string | undefined): Promise<void>;
- forwardInput(input: NetInput, forwardParams?: IMtcnnOptions): Promise<{
- results: MtcnnResult[];
- stats: any;
- }>;
- forward(input: TNetInput, forwardParams?: IMtcnnOptions): Promise<MtcnnResult[]>;
- forwardWithStats(input: TNetInput, forwardParams?: IMtcnnOptions): Promise<{
- results: MtcnnResult[];
- stats: any;
- }>;
- protected getDefaultModelName(): string;
- protected extractParamsFromWeigthMap(weightMap: tf.NamedTensorMap): {
- params: NetParams;
- paramMappings: import("../common").ParamMapping[];
- };
- protected extractParams(weights: Float32Array): {
- params: NetParams;
- paramMappings: import("../common").ParamMapping[];
- };
- }
|