PredictFaceExpressionsTask.d.ts 2.5 KB

123456789101112131415161718192021222324252627282930
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { TNetInput } from '../dom';
  3. import { WithFaceDetection } from '../factories/WithFaceDetection';
  4. import { WithFaceExpressions } from '../factories/WithFaceExpressions';
  5. import { WithFaceLandmarks } from '../factories/WithFaceLandmarks';
  6. import { ComposableTask } from './ComposableTask';
  7. import { ComputeAllFaceDescriptorsTask, ComputeSingleFaceDescriptorTask } from './ComputeFaceDescriptorsTasks';
  8. import { PredictAllAgeAndGenderTask, PredictAllAgeAndGenderWithFaceAlignmentTask, PredictSingleAgeAndGenderTask, PredictSingleAgeAndGenderWithFaceAlignmentTask } from './PredictAgeAndGenderTask';
  9. export declare class PredictFaceExpressionsTaskBase<TReturn, TParentReturn> extends ComposableTask<TReturn> {
  10. protected parentTask: ComposableTask<TParentReturn> | Promise<TParentReturn>;
  11. protected input: TNetInput;
  12. protected extractedFaces?: (tf.Tensor3D | HTMLCanvasElement)[] | undefined;
  13. constructor(parentTask: ComposableTask<TParentReturn> | Promise<TParentReturn>, input: TNetInput, extractedFaces?: (tf.Tensor3D | HTMLCanvasElement)[] | undefined);
  14. }
  15. export declare class PredictAllFaceExpressionsTask<TSource extends WithFaceDetection<{}>> extends PredictFaceExpressionsTaskBase<WithFaceExpressions<TSource>[], TSource[]> {
  16. run(): Promise<WithFaceExpressions<TSource>[]>;
  17. withAgeAndGender(): PredictAllAgeAndGenderTask<WithFaceExpressions<TSource>>;
  18. }
  19. export declare class PredictSingleFaceExpressionsTask<TSource extends WithFaceDetection<{}>> extends PredictFaceExpressionsTaskBase<WithFaceExpressions<TSource> | undefined, TSource | undefined> {
  20. run(): Promise<WithFaceExpressions<TSource> | undefined>;
  21. withAgeAndGender(): PredictSingleAgeAndGenderTask<WithFaceExpressions<TSource>>;
  22. }
  23. export declare class PredictAllFaceExpressionsWithFaceAlignmentTask<TSource extends WithFaceLandmarks<WithFaceDetection<{}>>> extends PredictAllFaceExpressionsTask<TSource> {
  24. withAgeAndGender(): PredictAllAgeAndGenderWithFaceAlignmentTask<WithFaceExpressions<TSource>>;
  25. withFaceDescriptors(): ComputeAllFaceDescriptorsTask<WithFaceExpressions<TSource>>;
  26. }
  27. export declare class PredictSingleFaceExpressionsWithFaceAlignmentTask<TSource extends WithFaceLandmarks<WithFaceDetection<{}>>> extends PredictSingleFaceExpressionsTask<TSource> {
  28. withAgeAndGender(): PredictSingleAgeAndGenderWithFaceAlignmentTask<WithFaceExpressions<TSource>>;
  29. withFaceDescriptor(): ComputeSingleFaceDescriptorTask<WithFaceExpressions<TSource>>;
  30. }