2020-12-23 18:58:47 +01:00
|
|
|
import * as tf from '../../dist/tfjs.esm';
|
2020-12-23 17:26:55 +01:00
|
|
|
import { ConvParams } from '../common/index';
|
|
|
|
|
export declare type ScaleLayerParams = {
|
|
|
|
|
weights: tf.Tensor1D;
|
|
|
|
|
biases: tf.Tensor1D;
|
|
|
|
|
};
|
|
|
|
|
export declare type ConvLayerParams = {
|
|
|
|
|
conv: ConvParams;
|
|
|
|
|
scale: ScaleLayerParams;
|
|
|
|
|
};
|
|
|
|
|
export declare type ResidualLayerParams = {
|
|
|
|
|
conv1: ConvLayerParams;
|
|
|
|
|
conv2: ConvLayerParams;
|
|
|
|
|
};
|
|
|
|
|
export declare type NetParams = {
|
|
|
|
|
conv32_down: ConvLayerParams;
|
|
|
|
|
conv32_1: ResidualLayerParams;
|
|
|
|
|
conv32_2: ResidualLayerParams;
|
|
|
|
|
conv32_3: ResidualLayerParams;
|
|
|
|
|
conv64_down: ResidualLayerParams;
|
|
|
|
|
conv64_1: ResidualLayerParams;
|
|
|
|
|
conv64_2: ResidualLayerParams;
|
|
|
|
|
conv64_3: ResidualLayerParams;
|
|
|
|
|
conv128_down: ResidualLayerParams;
|
|
|
|
|
conv128_1: ResidualLayerParams;
|
|
|
|
|
conv128_2: ResidualLayerParams;
|
|
|
|
|
conv256_down: ResidualLayerParams;
|
|
|
|
|
conv256_1: ResidualLayerParams;
|
|
|
|
|
conv256_2: ResidualLayerParams;
|
|
|
|
|
conv256_down_out: ResidualLayerParams;
|
|
|
|
|
fc: tf.Tensor2D;
|
|
|
|
|
};
|
