7 lines
322 B
TypeScript
7 lines
322 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>;
|
|
};
|
|
//# sourceMappingURL=boxPredictionLayer.d.ts.map
|