FaceProcessor.d.ts 1.5 KB

12345678910111213141516171819202122232425262728
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { NetInput } from '../dom';
  3. import { FaceFeatureExtractorParams, IFaceFeatureExtractor, TinyFaceFeatureExtractorParams } from '../faceFeatureExtractor/types';
  4. import { NeuralNetwork } from '../NeuralNetwork';
  5. import { NetParams } from './types';
  6. export declare abstract class FaceProcessor<TExtractorParams extends FaceFeatureExtractorParams | TinyFaceFeatureExtractorParams> extends NeuralNetwork<NetParams> {
  7. protected _faceFeatureExtractor: IFaceFeatureExtractor<TExtractorParams>;
  8. constructor(_name: string, faceFeatureExtractor: IFaceFeatureExtractor<TExtractorParams>);
  9. get faceFeatureExtractor(): IFaceFeatureExtractor<TExtractorParams>;
  10. protected abstract getDefaultModelName(): string;
  11. protected abstract getClassifierChannelsIn(): number;
  12. protected abstract getClassifierChannelsOut(): number;
  13. runNet(input: NetInput | tf.Tensor4D): tf.Tensor2D;
  14. dispose(throwOnRedispose?: boolean): void;
  15. loadClassifierParams(weights: Float32Array): void;
  16. extractClassifierParams(weights: Float32Array): {
  17. params: NetParams;
  18. paramMappings: import("../common").ParamMapping[];
  19. };
  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. }