Function tensorWith

tracker/test/net.test.ts:21–36

Build a zeroed YOLOX output tensor, then plant one airplane detection at * the given anchor index with explicit raw box regression values.

(
  anchorIdx: number,
  raw: { x: number; y: number; w: number; h: number; obj: number; cls: number },
  classId = 4,
)

/** Build a zeroed YOLOX output tensor, then plant one airplane detection at
 *  the given anchor index with explicit raw box regression values. */
function tensorWith(
  anchorIdx: number,
  raw: { x: number; y: number; w: number; h: number; obj: number; cls: number },
  classId = 4,
) {
  const A = anchorCount(INPUT);
  const data = new Float32Array(A * STEP);
  const o = anchorIdx * STEP;
  data[o] = raw.x;
  data[o + 1] = raw.y;
  data[o + 2] = raw.w;
  data[o + 3] = raw.h;
  data[o + 4] = raw.obj;
  data[o + 5 + classId] = raw.cls;
  return { data, dims: [1, A, STEP] };
}

describe("decodeYolox", () => {
  it("decodes a centered airplane anchor to the right pixel box", () => {