outputLayer.d.ts 272 B

123456
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { OutputLayerParams } from './types';
  3. export declare function outputLayer(boxPredictions: tf.Tensor4D, classPredictions: tf.Tensor4D, params: OutputLayerParams): {
  4. boxes: tf.Tensor2D[];
  5. scores: tf.Tensor1D[];
  6. };