TinyFaceDetector.d.ts 758 B

1234567891011121314151617
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { FaceDetection, Point } from '../classes';
  3. import { ParamMapping } from '../common';
  4. import { TNetInput } from '../dom';
  5. import { ITinyYolov2Options } from '../tinyYolov2';
  6. import { TinyYolov2Base } from '../tinyYolov2/TinyYolov2Base';
  7. import { TinyYolov2NetParams } from '../tinyYolov2/types';
  8. export declare class TinyFaceDetector extends TinyYolov2Base {
  9. constructor();
  10. get anchors(): Point[];
  11. locateFaces(input: TNetInput, forwardParams: ITinyYolov2Options): Promise<FaceDetection[]>;
  12. protected getDefaultModelName(): string;
  13. protected extractParamsFromWeigthMap(weightMap: tf.NamedTensorMap): {
  14. params: TinyYolov2NetParams;
  15. paramMappings: ParamMapping[];
  16. };
  17. }