6 lines
298 B
TypeScript
6 lines
298 B
TypeScript
import * as tf from '@tensorflow/tfjs-core';
|
|
import { OutputLayerParams } from './types';
|
|
export declare function outputLayer(boxPredictions: tf.Tensor4D, classPredictions: tf.Tensor4D, params: OutputLayerParams): {
|
|
boxes: tf.Tensor<tf.Rank.R2>[];
|
|
scores: tf.Tensor<tf.Rank.R1>[];
|
|
};
|