2020-12-23 18:58:47 +01:00
|
|
|
import * as tf from '../../dist/tfjs.esm';
|
2020-12-23 17:26:55 +01:00
|
|
|
import { NetInput, TNetInput } from '../dom/index';
|
|
|
|
import { NeuralNetwork } from '../NeuralNetwork';
|
|
|
|
import { TinyXceptionParams } from './types';
|
|
|
|
export declare class TinyXception extends NeuralNetwork<TinyXceptionParams> {
|
|
|
|
private _numMainBlocks;
|
|
|
|
constructor(numMainBlocks: number);
|
|
|
|
forwardInput(input: NetInput): tf.Tensor4D;
|
|
|
|
forward(input: TNetInput): Promise<tf.Tensor4D>;
|
|
|
|
protected getDefaultModelName(): string;
|
2021-01-12 16:14:33 +01:00
|
|
|
protected extractParamsFromWeightMap(weightMap: tf.NamedTensorMap): {
|
2020-12-23 17:26:55 +01:00
|
|
|
params: TinyXceptionParams;
|
2020-12-23 18:58:47 +01:00
|
|
|
paramMappings: import("../common/types").ParamMapping[];
|
2020-12-23 17:26:55 +01:00
|
|
|
};
|
|
|
|
protected extractParams(weights: Float32Array): {
|
|
|
|
params: TinyXceptionParams;
|
2020-12-23 18:58:47 +01:00
|
|
|
paramMappings: import("../common/types").ParamMapping[];
|
2020-12-23 17:26:55 +01:00
|
|
|
};
|
|
|
|
}
|