FaceFeatureExtractor.d.ts 888 B

123456789101112131415161718
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { NetInput, TNetInput } from '../dom';
  3. import { NeuralNetwork } from '../NeuralNetwork';
  4. import { FaceFeatureExtractorParams, IFaceFeatureExtractor } from './types';
  5. export declare class FaceFeatureExtractor extends NeuralNetwork<FaceFeatureExtractorParams> implements IFaceFeatureExtractor<FaceFeatureExtractorParams> {
  6. constructor();
  7. forwardInput(input: NetInput): tf.Tensor4D;
  8. forward(input: TNetInput): Promise<tf.Tensor4D>;
  9. protected getDefaultModelName(): string;
  10. protected extractParamsFromWeigthMap(weightMap: tf.NamedTensorMap): {
  11. params: FaceFeatureExtractorParams;
  12. paramMappings: import("../common").ParamMapping[];
  13. };
  14. protected extractParams(weights: Float32Array): {
  15. params: FaceFeatureExtractorParams;
  16. paramMappings: import("../common").ParamMapping[];
  17. };
  18. }