7 lines
313 B
TypeScript
7 lines
313 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.Tensor2D[];
|
|
scores: tf.Tensor1D[];
|
|
};
|
|
//# sourceMappingURL=outputLayer.d.ts.map
|