Function createBatchDetTensor

paddleocr-js/packages/core/src/models/det.ts:348–373

(
  ort: OrtModule,
  preps: DetPreprocessResult[],
  maxH: number,
  maxW: number
)

}

function createBatchDetTensor(
  ort: OrtModule,
  preps: DetPreprocessResult[],
  maxH: number,
  maxW: number
): Tensor {
  const batch = preps.length;
  const plane = 3 * maxH * maxW;
  const out = new Float32Array(batch * plane);
  for (let i = 0; i < batch; i += 1) {
    const prep = preps[i];
    const chw = prep.tensor.data as Float32Array;
    const { dstH, dstW } = prep;
    const base = i * plane;
    for (let c = 0; c < 3; c += 1) {
      const srcChannelBase = c * dstH * dstW;
      const dstChannelBase = base + c * maxH * maxW;
      for (let y = 0; y < dstH; y += 1) {
        const srcRow = srcChannelBase + y * dstW;
        const dstRow = dstChannelBase + y * maxW;
        out.set(chw.subarray(srcRow, srcRow + dstW), dstRow);
      }
    }
  }
  return new ort.Tensor("float32", out, [batch, 3, maxH, maxW]);
}

function packDetBatchTensor(ort: OrtModule, preps: DetPreprocessResult[]): Tensor {
  const maxH = Math.max(...preps.map((p) => p.dstH));