import * as tf from '@tensorflow/tfjs-core';
|
import { boxPredictionLayer } from './boxPredictionLayer';
|
import { pointwiseConvLayer } from './pointwiseConvLayer';
|
export function predictionLayer(x, conv11, params) {
|
return tf.tidy(function () {
|
var conv0 = pointwiseConvLayer(x, params.conv_0, [1, 1]);
|
var conv1 = pointwiseConvLayer(conv0, params.conv_1, [2, 2]);
|
var conv2 = pointwiseConvLayer(conv1, params.conv_2, [1, 1]);
|
var conv3 = pointwiseConvLayer(conv2, params.conv_3, [2, 2]);
|
var conv4 = pointwiseConvLayer(conv3, params.conv_4, [1, 1]);
|
var conv5 = pointwiseConvLayer(conv4, params.conv_5, [2, 2]);
|
var conv6 = pointwiseConvLayer(conv5, params.conv_6, [1, 1]);
|
var conv7 = pointwiseConvLayer(conv6, params.conv_7, [2, 2]);
|
var boxPrediction0 = boxPredictionLayer(conv11, params.box_predictor_0);
|
var boxPrediction1 = boxPredictionLayer(x, params.box_predictor_1);
|
var boxPrediction2 = boxPredictionLayer(conv1, params.box_predictor_2);
|
var boxPrediction3 = boxPredictionLayer(conv3, params.box_predictor_3);
|
var boxPrediction4 = boxPredictionLayer(conv5, params.box_predictor_4);
|
var boxPrediction5 = boxPredictionLayer(conv7, params.box_predictor_5);
|
var boxPredictions = tf.concat([
|
boxPrediction0.boxPredictionEncoding,
|
boxPrediction1.boxPredictionEncoding,
|
boxPrediction2.boxPredictionEncoding,
|
boxPrediction3.boxPredictionEncoding,
|
boxPrediction4.boxPredictionEncoding,
|
boxPrediction5.boxPredictionEncoding
|
], 1);
|
var classPredictions = tf.concat([
|
boxPrediction0.classPrediction,
|
boxPrediction1.classPrediction,
|
boxPrediction2.classPrediction,
|
boxPrediction3.classPrediction,
|
boxPrediction4.classPrediction,
|
boxPrediction5.classPrediction
|
], 1);
|
return {
|
boxPredictions: boxPredictions,
|
classPredictions: classPredictions
|
};
|
});
|
}
|
//# sourceMappingURL=predictionLayer.js.map
|
