SsdMobilenetv1.d.ts 1.0 KB

123456789101112131415161718192021222324252627
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { FaceDetection } from '../classes/FaceDetection';
  3. import { NetInput, TNetInput } from '../dom';
  4. import { NeuralNetwork } from '../NeuralNetwork';
  5. import { ISsdMobilenetv1Options } from './SsdMobilenetv1Options';
  6. import { NetParams } from './types';
  7. export declare class SsdMobilenetv1 extends NeuralNetwork<NetParams> {
  8. constructor();
  9. forwardInput(input: NetInput): {
  10. boxes: tf.Tensor2D[];
  11. scores: tf.Tensor1D[];
  12. };
  13. forward(input: TNetInput): Promise<{
  14. boxes: tf.Tensor2D[];
  15. scores: tf.Tensor1D[];
  16. }>;
  17. locateFaces(input: TNetInput, options?: ISsdMobilenetv1Options): Promise<FaceDetection[]>;
  18. protected getDefaultModelName(): string;
  19. protected extractParamsFromWeigthMap(weightMap: tf.NamedTensorMap): {
  20. params: NetParams;
  21. paramMappings: import("../common").ParamMapping[];
  22. };
  23. protected extractParams(weights: Float32Array): {
  24. params: NetParams;
  25. paramMappings: import("../common").ParamMapping[];
  26. };
  27. }