AgeGenderNet.d.ts 1.3 KB

1234567891011121314151617181920212223242526272829
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { TinyXception } from '../xception/TinyXception';
  3. import { AgeAndGenderPrediction, NetOutput, NetParams } from './types';
  4. import { NeuralNetwork } from '../NeuralNetwork';
  5. import { NetInput, TNetInput } from '../dom';
  6. export declare class AgeGenderNet extends NeuralNetwork<NetParams> {
  7. private _faceFeatureExtractor;
  8. constructor(faceFeatureExtractor?: TinyXception);
  9. get faceFeatureExtractor(): TinyXception;
  10. runNet(input: NetInput | tf.Tensor4D): NetOutput;
  11. forwardInput(input: NetInput | tf.Tensor4D): NetOutput;
  12. forward(input: TNetInput): Promise<NetOutput>;
  13. predictAgeAndGender(input: TNetInput): Promise<AgeAndGenderPrediction | AgeAndGenderPrediction[]>;
  14. protected getDefaultModelName(): string;
  15. dispose(throwOnRedispose?: boolean): void;
  16. loadClassifierParams(weights: Float32Array): void;
  17. extractClassifierParams(weights: Float32Array): {
  18. params: NetParams;
  19. paramMappings: import("../common").ParamMapping[];
  20. };
  21. protected extractParamsFromWeigthMap(weightMap: tf.NamedTensorMap): {
  22. params: NetParams;
  23. paramMappings: import("../common").ParamMapping[];
  24. };
  25. protected extractParams(weights: Float32Array): {
  26. params: NetParams;
  27. paramMappings: import("../common").ParamMapping[];
  28. };
  29. }