41 lines
952 B
TypeScript
41 lines
952 B
TypeScript
import * as tf from '../../dist/tfjs.esm';
|
|
|
|
import { ConvParams } from '../common/index';
|
|
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
|