Mtcnn.d.ts 1.2 KB

12345678910111213141516171819202122232425262728
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { NetInput, TNetInput } from '../dom';
  3. import { NeuralNetwork } from '../NeuralNetwork';
  4. import { IMtcnnOptions } from './MtcnnOptions';
  5. import { MtcnnResult, NetParams } from './types';
  6. export declare class Mtcnn extends NeuralNetwork<NetParams> {
  7. constructor();
  8. load(weightsOrUrl: Float32Array | string | undefined): Promise<void>;
  9. loadFromDisk(filePath: string | undefined): Promise<void>;
  10. forwardInput(input: NetInput, forwardParams?: IMtcnnOptions): Promise<{
  11. results: MtcnnResult[];
  12. stats: any;
  13. }>;
  14. forward(input: TNetInput, forwardParams?: IMtcnnOptions): Promise<MtcnnResult[]>;
  15. forwardWithStats(input: TNetInput, forwardParams?: IMtcnnOptions): Promise<{
  16. results: MtcnnResult[];
  17. stats: any;
  18. }>;
  19. protected getDefaultModelName(): string;
  20. protected extractParamsFromWeigthMap(weightMap: tf.NamedTensorMap): {
  21. params: NetParams;
  22. paramMappings: import("../common").ParamMapping[];
  23. };
  24. protected extractParams(weights: Float32Array): {
  25. params: NetParams;
  26. paramMappings: import("../common").ParamMapping[];
  27. };
  28. }