TinyXception.d.ts 813 B

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