UNPKG

tfjs-tiny-yolov2

Version:

Tiny YOLO v2 object detection with tensorflow.js.

16 lines (15 loc) 468 B
import * as tf from '@tensorflow/tfjs-core'; export declare type ConvParams = { filters: tf.Tensor4D; bias: tf.Tensor1D; }; export declare type FCParams = { weights: tf.Tensor2D; bias: tf.Tensor1D; }; export declare class SeparableConvParams { depthwise_filter: tf.Tensor4D; pointwise_filter: tf.Tensor4D; bias: tf.Tensor1D; constructor(depthwise_filter: tf.Tensor4D, pointwise_filter: tf.Tensor4D, bias: tf.Tensor1D); }