1234567891011121314151617181920212223242526272829 |
- import * as tf from '@tensorflow/tfjs-core';
- import { TinyXception } from '../xception/TinyXception';
- import { AgeAndGenderPrediction, NetOutput, NetParams } from './types';
- import { NeuralNetwork } from '../NeuralNetwork';
- import { NetInput, TNetInput } from '../dom';
- export declare class AgeGenderNet extends NeuralNetwork<NetParams> {
- private _faceFeatureExtractor;
- constructor(faceFeatureExtractor?: TinyXception);
- get faceFeatureExtractor(): TinyXception;
- runNet(input: NetInput | tf.Tensor4D): NetOutput;
- forwardInput(input: NetInput | tf.Tensor4D): NetOutput;
- forward(input: TNetInput): Promise<NetOutput>;
- predictAgeAndGender(input: TNetInput): Promise<AgeAndGenderPrediction | AgeAndGenderPrediction[]>;
- protected getDefaultModelName(): string;
- dispose(throwOnRedispose?: boolean): void;
- loadClassifierParams(weights: Float32Array): void;
- extractClassifierParams(weights: Float32Array): {
- params: NetParams;
- paramMappings: import("../common").ParamMapping[];
- };
- protected extractParamsFromWeigthMap(weightMap: tf.NamedTensorMap): {
- params: NetParams;
- paramMappings: import("../common").ParamMapping[];
- };
- protected extractParams(weights: Float32Array): {
- params: NetParams;
- paramMappings: import("../common").ParamMapping[];
- };
- }
|