1
2
3
4
5
6
7
8
9
10
11
12
13
14
| import * as tf from '@tensorflow/tfjs-core';
| import { convLayer } from '../common';
| export function boxPredictionLayer(x, params) {
| return tf.tidy(function () {
| var batchSize = x.shape[0];
| var boxPredictionEncoding = tf.reshape(convLayer(x, params.box_encoding_predictor), [batchSize, -1, 1, 4]);
| var classPrediction = tf.reshape(convLayer(x, params.class_predictor), [batchSize, -1, 3]);
| return {
| boxPredictionEncoding: boxPredictionEncoding,
| classPrediction: classPrediction
| };
| });
| }
| //# sourceMappingURL=boxPredictionLayer.js.map
|
|
