40 lines
947 B
TypeScript
40 lines
947 B
TypeScript
|
import * as tf from '@tensorflow/tfjs-core';
|
||
|
|
|
||
|
|
import { ConvParams } from '../common';
|
||
|
|
import { SeparableConvParams } from '../common/types';
|
||
|
|
|
||
|
|
export type BatchNorm = {
|
||
|
|
sub: tf.Tensor1D
|
||
|
|
truediv: tf.Tensor1D
|
||
|
|
}
|
||
|
|
|
||
|
|
export type ConvWithBatchNorm = {
|
||
|
|
conv: ConvParams
|
||
|
|
bn: BatchNorm
|
||
|
|
}
|
||
|
|
|
||
|
|
export type MobilenetParams = {
|
||
|
|
conv0: SeparableConvParams | ConvParams
|
||
|
|
conv1: SeparableConvParams
|
||
|
|
conv2: SeparableConvParams
|
||
|
|
conv3: SeparableConvParams
|
||
|
|
conv4: SeparableConvParams
|
||
|
|
conv5: SeparableConvParams
|
||
|
|
conv6?: SeparableConvParams
|
||
|
|
conv7?: SeparableConvParams
|
||
|
|
conv8: ConvParams
|
||
|
|
}
|
||
|
|
|
||
|
|
export type DefaultTinyYolov2NetParams = {
|
||
|
|
conv0: ConvWithBatchNorm
|
||
|
|
conv1: ConvWithBatchNorm
|
||
|
|
conv2: ConvWithBatchNorm
|
||
|
|
conv3: ConvWithBatchNorm
|
||
|
|
conv4: ConvWithBatchNorm
|
||
|
|
conv5: ConvWithBatchNorm
|
||
|
|
conv6: ConvWithBatchNorm
|
||
|
|
conv7: ConvWithBatchNorm
|
||
|
|
conv8: ConvParams
|
||
|
|
}
|
||
|
|
|
||
|
|
export type TinyYolov2NetParams = DefaultTinyYolov2NetParams | MobilenetParams
|