TinyYolov2.d.ts 813 B

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