Function decodeBBox

lib/fr_detect.js:202–248

(bbox, priors, onnx_config)

}

function decodeBBox(bbox, priors, onnx_config) {
  var variances = onnx_config.variance
  var loc = ndarray(bbox.data, [4420, 4]);
  // console.log(bbox, priors);
  var before_prior = priors.hi(null, 2),
      after_prior = priors.lo(null, 2);

  var before_loc = loc.hi(null, 2),
      after_loc = loc.lo(null, 2);

  var before_result = ndarray(new Float32Array(before_loc.shape[0] * before_loc.shape[1]), [before_loc.shape[0], 2]);
  var before_temp = ndarray(new Float32Array(before_loc.shape[0] * before_loc.shape[1]), [before_loc.shape[0], 2]);
  var before_temp2 = ndarray(new Float32Array(before_loc.shape[0] * before_loc.shape[1]), [before_loc.shape[0], 2]);

  var after_result = ndarray(new Float32Array(before_loc.shape[0] * before_loc.shape[1]), [before_loc.shape[0], 2]);
  var after_temp = ndarray(new Float32Array(before_loc.shape[0] * before_loc.shape[1]), [before_loc.shape[0], 2]);
  var after_temp2 = ndarray(new Float32Array(before_loc.shape[0] * before_loc.shape[1]), [before_loc.shape[0], 2]);
  var after_temp3 = ndarray(new Float32Array(before_loc.shape[0] * before_loc.shape[1]), [before_loc.shape[0], 2]);
  var after_temp4 = ndarray(new Float32Array(before_loc.shape[0] * before_loc.shape[1]), [before_loc.shape[0], 2]);

  var boxes = ndarray(new Float32Array(before_loc.shape[0] * 4), [before_loc.shape[0], 4]);

  // Before
  ops.mul(before_temp, before_loc, after_prior);
  ops.muls(before_temp2, before_temp, variances[0]);
  ops.add(before_result, before_temp2, before_prior);

  // After
  ops.muls(after_temp, after_loc, variances[1]);
  ops.exp(after_temp2, after_temp);
  ops.mul(after_temp3, after_temp2, after_prior);

  for (var index = 0; index < 4; index++)
    ops.assign(after_result.pick(null, index), after_temp3.pick(null, index));

  ops.divs(after_temp4, after_temp3, -2);
  ops.addeq(before_result, after_temp4);

  ops.addeq(after_result, before_result);

  ops.assign(boxes.pick(null, 0), before_result.pick(null, 0));
  ops.assign(boxes.pick(null, 1), before_result.pick(null, 1));
  ops.assign(boxes.pick(null, 2), after_result.pick(null, 0));
  ops.assign(boxes.pick(null, 3), after_result.pick(null, 1));

  return boxes;
}

function scaleMultiplyBBox(boxes_arr, scale) {
  var total_result = boxes_arr.shape[0];