6 lines
280 B
TypeScript
6 lines
280 B
TypeScript
import * as tf from '@tensorflow/tfjs-core';
|
|
import { BoxPredictionParams } from './types';
|
|
export declare function boxPredictionLayer(x: tf.Tensor4D, params: BoxPredictionParams): {
|
|
boxPredictionEncoding: tf.Tensor<tf.Rank>;
|
|
classPrediction: tf.Tensor<tf.Rank>;
|
|
};
|