@mauriciobenjamin700/ort-vision-sdk-web 0.2.1

package/dist/postprocess/detection.js

@@ -0,0 +1,283 @@
+/**
+ * Detection head postprocessing: anchor-free YOLO decoding + non-maximum suppression.
+ *
+ * The shared {@link decodeYoloAnchors} helper does the per-anchor work that
+ * is identical for plain detection and segmentation (transpose, xywh→xyxy,
+ * letterbox unmap, per-class NMS, sort & cap). {@link decodeYolo} is a thin
+ * wrapper around it; the segmentation module ({@link ./segmentation.js})
+ * calls the helper directly so it can also recover the per-anchor mask
+ * coefficients.
+ *
+ * Works for any YOLO export with the post-v8 anchor-free head:
+ * **YOLOv8 / v9 / v10 / v11 / v12** detect heads, all of which share the
+ * `[1, 4 + nc, N]` output layout.
+ */
+import { BoundingBox } from "../types.js";
+/**
+ * Greedy non-maximum suppression on axis-aligned bounding boxes.
+ *
+ * Mirrors `torchvision.ops.nms` (keeps boxes with the highest score, drops
+ * any subsequent box whose IoU exceeds the threshold).
+ *
+ * @param boxes Flat array of length `4 * N` in xyxy order: `[x1,y1,x2,y2, ...]`.
+ * @param scores Detection score per box, length `N`.
+ * @param iouThreshold Boxes with IoU above this threshold relative to a kept box are suppressed.
+ * @returns Indices of kept boxes, in descending score order.
+ */
+export function nms(boxes, scores, iouThreshold) {
+    const n = scores.length;
+    if (n === 0)
+        return new Int32Array(0);
+    const areas = new Float32Array(n);
+    for (let i = 0; i < n; i++) {
+        const x1 = boxes[i * 4];
+        const y1 = boxes[i * 4 + 1];
+        const x2 = boxes[i * 4 + 2];
+        const y2 = boxes[i * 4 + 3];
+        areas[i] = Math.max(0, x2 - x1) * Math.max(0, y2 - y1);
+    }
+    const order = new Array(n);
+    for (let i = 0; i < n; i++)
+        order[i] = i;
+    order.sort((a, b) => scores[b] - scores[a]);
+    const suppressed = new Uint8Array(n);
+    const keep = [];
+    for (let oi = 0; oi < order.length; oi++) {
+        const i = order[oi];
+        if (suppressed[i])
+            continue;
+        keep.push(i);
+        const ax1 = boxes[i * 4];
+        const ay1 = boxes[i * 4 + 1];
+        const ax2 = boxes[i * 4 + 2];
+        const ay2 = boxes[i * 4 + 3];
+        const ai = areas[i];
+        for (let oj = oi + 1; oj < order.length; oj++) {
+            const j = order[oj];
+            if (suppressed[j])
+                continue;
+            const bx1 = boxes[j * 4];
+            const by1 = boxes[j * 4 + 1];
+            const bx2 = boxes[j * 4 + 2];
+            const by2 = boxes[j * 4 + 3];
+            const ix1 = Math.max(ax1, bx1);
+            const iy1 = Math.max(ay1, by1);
+            const ix2 = Math.min(ax2, bx2);
+            const iy2 = Math.min(ay2, by2);
+            const iw = Math.max(0, ix2 - ix1);
+            const ih = Math.max(0, iy2 - iy1);
+            const inter = iw * ih;
+            const union = ai + areas[j] - inter;
+            const iou = union > 0 ? inter / union : 0;
+            if (iou > iouThreshold)
+                suppressed[j] = 1;
+        }
+    }
+    return Int32Array.from(keep);
+}
+/**
+ * Per-class NMS — boxes are suppressed only by other boxes of the same class.
+ *
+ * Mirrors `torchvision.ops.batched_nms`.
+ *
+ * @param boxes Flat array of length `4 * N` in xyxy order.
+ * @param scores Detection score per box, length `N`.
+ * @param idxs Class index per box, length `N`. Boxes with different `idxs`
+ *   never suppress each other.
+ * @param iouThreshold IoU threshold for suppression within a class.
+ */
+export function batchedNms(boxes, scores, idxs, iouThreshold) {
+    if (scores.length === 0)
+        return new Int32Array(0);
+    const byClass = new Map();
+    for (let i = 0; i < idxs.length; i++) {
+        const c = idxs[i];
+        const list = byClass.get(c);
+        if (list === undefined)
+            byClass.set(c, [i]);
+        else
+            list.push(i);
+    }
+    const keep = [];
+    for (const indices of byClass.values()) {
+        const m = indices.length;
+        const subBoxes = new Float32Array(m * 4);
+        const subScores = new Float32Array(m);
+        for (let k = 0; k < m; k++) {
+            const i = indices[k];
+            subBoxes[k * 4] = boxes[i * 4];
+            subBoxes[k * 4 + 1] = boxes[i * 4 + 1];
+            subBoxes[k * 4 + 2] = boxes[i * 4 + 2];
+            subBoxes[k * 4 + 3] = boxes[i * 4 + 3];
+            subScores[k] = scores[i];
+        }
+        const subKeep = nms(subBoxes, subScores, iouThreshold);
+        for (let k = 0; k < subKeep.length; k++) {
+            keep.push(indices[subKeep[k]]);
+        }
+    }
+    keep.sort((a, b) => scores[b] - scores[a]);
+    return Int32Array.from(keep);
+}
+/**
+ * Shared YOLO per-anchor decode used by both detection and segmentation
+ * (v8 / v9 / v10 / v11 / v12).
+ *
+ * Only the first `4 + numClasses` channels are read; later channels (e.g.
+ * mask coefficients) are ignored — callers can fetch them via the returned
+ * {@link DecodedAnchors.anchorIndices}.
+ *
+ * @param data Flat per-anchor output, length `channels * numAnchors`.
+ * @param dims Dims as reported by ORT, e.g. `[1, 84, 8400]` (det) or
+ *   `[1, 116, 8400]` (seg). The leading batch dim must be 1.
+ */
+export function decodeYoloAnchors(data, dims, options) {
+    let normalized = dims;
+    if (normalized.length === 3) {
+        if (normalized[0] !== 1) {
+            throw new Error(`decodeYoloAnchors: expected batch size 1, got ${normalized[0]}.`);
+        }
+        normalized = [normalized[1], normalized[2]];
+    }
+    if (normalized.length !== 2) {
+        throw new Error(`decodeYoloAnchors: expected 2-D output after batch removal, got dims=${JSON.stringify(dims)}.`);
+    }
+    const channels = normalized[0];
+    const numAnchors = normalized[1];
+    const { numClasses, originalWidth, originalHeight, padLeft, padTop, scale, confThreshold, iouThreshold, maxDetections, } = options;
+    if (numClasses < 1 || numClasses + 4 > channels) {
+        throw new Error(`decodeYoloAnchors: invalid numClasses=${numClasses} for channels=${channels}.`);
+    }
+    if (data.length !== channels * numAnchors) {
+        throw new Error(`decodeYoloAnchors: data length ${data.length} does not match channels*numAnchors=${channels * numAnchors}.`);
+    }
+    const candidates = [];
+    for (let a = 0; a < numAnchors; a++) {
+        let bestCls = 0;
+        let bestScore = -Infinity;
+        for (let c = 0; c < numClasses; c++) {
+            const s = data[(4 + c) * numAnchors + a];
+            if (s !== undefined && s > bestScore) {
+                bestScore = s;
+                bestCls = c;
+            }
+        }
+        if (bestScore < confThreshold)
+            continue;
+        const cx = data[a];
+        const cy = data[numAnchors + a];
+        const w = data[2 * numAnchors + a];
+        const h = data[3 * numAnchors + a];
+        let x1 = cx - w / 2;
+        let y1 = cy - h / 2;
+        let x2 = cx + w / 2;
+        let y2 = cy + h / 2;
+        x1 = (x1 - padLeft) / scale;
+        y1 = (y1 - padTop) / scale;
+        x2 = (x2 - padLeft) / scale;
+        y2 = (y2 - padTop) / scale;
+        x1 = Math.max(0, Math.min(originalWidth, x1));
+        y1 = Math.max(0, Math.min(originalHeight, y1));
+        x2 = Math.max(0, Math.min(originalWidth, x2));
+        y2 = Math.max(0, Math.min(originalHeight, y2));
+        candidates.push({ anchorIdx: a, x1, y1, x2, y2, classId: bestCls, confidence: bestScore });
+    }
+    if (candidates.length === 0)
+        return emptyDecoded();
+    // Build flat arrays then delegate to batchedNms — same algorithm as before
+    // but funnelled through the public per-class NMS helper.
+    const flatBoxes = new Float32Array(candidates.length * 4);
+    const scoresArr = new Float32Array(candidates.length);
+    const idxsArr = new Int32Array(candidates.length);
+    for (let i = 0; i < candidates.length; i++) {
+        const c = candidates[i];
+        flatBoxes[i * 4] = c.x1;
+        flatBoxes[i * 4 + 1] = c.y1;
+        flatBoxes[i * 4 + 2] = c.x2;
+        flatBoxes[i * 4 + 3] = c.y2;
+        scoresArr[i] = c.confidence;
+        idxsArr[i] = c.classId;
+    }
+    const kept = batchedNms(flatBoxes, scoresArr, idxsArr, iouThreshold);
+    if (kept.length === 0)
+        return emptyDecoded();
+    const limited = Array.from(kept).slice(0, maxDetections);
+    const k = limited.length;
+    const anchorIndices = new Int32Array(k);
+    const boxesXyxy = new Float32Array(k * 4);
+    const classIds = new Int32Array(k);
+    const confidences = new Float32Array(k);
+    for (let i = 0; i < k; i++) {
+        const c = candidates[limited[i]];
+        anchorIndices[i] = c.anchorIdx;
+        boxesXyxy[i * 4] = c.x1;
+        boxesXyxy[i * 4 + 1] = c.y1;
+        boxesXyxy[i * 4 + 2] = c.x2;
+        boxesXyxy[i * 4 + 3] = c.y2;
+        classIds[i] = c.classId;
+        confidences[i] = c.confidence;
+    }
+    return { anchorIndices, boxesXyxy, classIds, confidences };
+}
+function emptyDecoded() {
+    return {
+        anchorIndices: new Int32Array(0),
+        boxesXyxy: new Float32Array(0),
+        classIds: new Int32Array(0),
+        confidences: new Float32Array(0),
+    };
+}
+/**
+ * Decode an anchor-free YOLO detection output into a list of detections.
+ *
+ * Works for **YOLOv8 / v9 / v10 / v11 / v12** detect heads.
+ *
+ * Expected raw shape: `[1, 4 + numClasses, N]`. `numClasses` is inferred
+ * from the channel count.
+ */
+export function decodeYolo(output, outputDims, options) {
+    const channels = outputDims.length === 3 ? outputDims[1] : outputDims[0];
+    if (channels === undefined || channels < 5) {
+        throw new Error(`decodeYolo: invalid output channel count ${channels} (expected >= 5).`);
+    }
+    const numClasses = channels - 4;
+    const decoded = decodeYoloAnchors(output, outputDims, {
+        numClasses,
+        ...options,
+    });
+    const results = [];
+    for (let i = 0; i < decoded.classIds.length; i++) {
+        results.push({
+            bbox: new BoundingBox(decoded.boxesXyxy[i * 4], decoded.boxesXyxy[i * 4 + 1], decoded.boxesXyxy[i * 4 + 2], decoded.boxesXyxy[i * 4 + 3]),
+            classId: decoded.classIds[i],
+            confidence: decoded.confidences[i],
+        });
+    }
+    return results;
+}
+let _warnedDecodeYoloV8 = false;
+let _warnedDecodeYoloV8Anchors = false;
+/**
+ * @deprecated since 0.2.0 — use {@link decodeYolo}. Same behavior; the
+ * decoder covers v8/v9/v10/v11/v12 detect heads. Will be removed in 0.3.0.
+ */
+export function decodeYoloV8(output, outputDims, options) {
+    if (!_warnedDecodeYoloV8) {
+        _warnedDecodeYoloV8 = true;
+        console.warn("[@ort-vision-sdk/web] decodeYoloV8 is deprecated since 0.2.0; use decodeYolo. " +
+            "The alias will be removed in 0.3.0.");
+    }
+    return decodeYolo(output, outputDims, options);
+}
+/**
+ * @deprecated since 0.2.0 — use {@link decodeYoloAnchors}. Will be removed in 0.3.0.
+ */
+export function decodeYoloV8Anchors(data, dims, options) {
+    if (!_warnedDecodeYoloV8Anchors) {
+        _warnedDecodeYoloV8Anchors = true;
+        console.warn("[@ort-vision-sdk/web] decodeYoloV8Anchors is deprecated since 0.2.0; use decodeYoloAnchors. " +
+            "The alias will be removed in 0.3.0.");
+    }
+    return decodeYoloAnchors(data, dims, options);
+}
+//# sourceMappingURL=detection.js.map

package/dist/postprocess/detection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"detection.js","sourceRoot":"","sources":["../../src/postprocess/detection.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C;;;;;;;;;;GAUG;AACH,MAAM,UAAU,GAAG,CACjB,KAAmB,EACnB,MAAoB,EACpB,YAAoB;IAEpB,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;IACxB,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IAEtC,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAW,CAAC;QAClC,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;QACtC,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;QACtC,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;QACtC,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;IACzD,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,KAAK,CAAS,CAAC,CAAC,CAAC;IACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;QAAE,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACzC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAE,MAAM,CAAC,CAAC,CAAY,GAAI,MAAM,CAAC,CAAC,CAAY,CAAC,CAAC;IAEpE,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IACrC,MAAM,IAAI,GAAa,EAAE,CAAC;IAE1B,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,KAAK,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC;QACzC,MAAM,CAAC,GAAG,KAAK,CAAC,EAAE,CAAW,CAAC;QAC9B,IAAI,UAAU,CAAC,CAAC,CAAC;YAAE,SAAS;QAC5B,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEb,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAW,CAAC;QACnC,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;QACvC,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;QACvC,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;QACvC,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,CAAW,CAAC;QAE9B,KAAK,IAAI,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,KAAK,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC;YAC9C,MAAM,CAAC,GAAG,KAAK,CAAC,EAAE,CAAW,CAAC;YAC9B,IAAI,UAAU,CAAC,CAAC,CAAC;gBAAE,SAAS;YAE5B,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAW,CAAC;YACnC,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;YACvC,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;YACvC,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;YAEvC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC/B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC/B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC/B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC/B,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,GAAG,GAAG,CAAC,CAAC;YAClC,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,GAAG,GAAG,CAAC,CAAC;YAClC,MAAM,KAAK,GAAG,EAAE,GAAG,EAAE,CAAC;YACtB,MAAM,KAAK,GAAG,EAAE,GAAI,KAAK,CAAC,CAAC,CAAY,GAAG,KAAK,CAAC;YAChD,MAAM,GAAG,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1C,IAAI,GAAG,GAAG,YAAY;gBAAE,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IAED,OAAO,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC/B,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAM,UAAU,UAAU,CACxB,KAAmB,EACnB,MAAoB,EACpB,IAAgB,EAChB,YAAoB;IAEpB,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IAElD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC5C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACrC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;QAC5B,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,IAAI,KAAK,SAAS;YAAE,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;;YACvC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;IAED,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;QACvC,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;QACzB,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACzC,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAW,CAAC;YAC/B,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAW,CAAC;YACzC,QAAQ,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;YACjD,QAAQ,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;YACjD,QAAQ,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;YACjD,SAAS,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAW,CAAC;QACrC,CAAC;QACD,MAAM,OAAO,GAAG,GAAG,CAAC,QAAQ,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC;QACvD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAW,CAAW,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAE,MAAM,CAAC,CAAC,CAAY,GAAI,MAAM,CAAC,CAAC,CAAY,CAAC,CAAC;IACnE,OAAO,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC/B,CAAC;AA0BD;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,iBAAiB,CAC/B,IAAkB,EAClB,IAAuB,EACvB,OAAiC;IAEjC,IAAI,UAAU,GAAG,IAAI,CAAC;IACtB,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,IAAI,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,iDAAiD,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACrF,CAAC;QACD,UAAU,GAAG,CAAC,UAAU,CAAC,CAAC,CAAW,EAAE,UAAU,CAAC,CAAC,CAAW,CAAC,CAAC;IAClE,CAAC;IACD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CACb,wEAAwE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAChG,CAAC;IACJ,CAAC;IACD,MAAM,QAAQ,GAAG,UAAU,CAAC,CAAC,CAAW,CAAC;IACzC,MAAM,UAAU,GAAG,UAAU,CAAC,CAAC,CAAW,CAAC;IAE3C,MAAM,EACJ,UAAU,EACV,aAAa,EACb,cAAc,EACd,OAAO,EACP,MAAM,EACN,KAAK,EACL,aAAa,EACb,YAAY,EACZ,aAAa,GACd,GAAG,OAAO,CAAC;IAEZ,IAAI,UAAU,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,GAAG,QAAQ,EAAE,CAAC;QAChD,MAAM,IAAI,KAAK,CACb,yCAAyC,UAAU,iBAAiB,QAAQ,GAAG,CAChF,CAAC;IACJ,CAAC;IACD,IAAI,IAAI,CAAC,MAAM,KAAK,QAAQ,GAAG,UAAU,EAAE,CAAC;QAC1C,MAAM,IAAI,KAAK,CACb,kCAAkC,IAAI,CAAC,MAAM,uCAAuC,QAAQ,GAAG,UAAU,GAAG,CAC7G,CAAC;IACJ,CAAC;IAWD,MAAM,UAAU,GAAgB,EAAE,CAAC;IAEnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,SAAS,GAAG,CAAC,QAAQ,CAAC;QAC1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,UAAU,GAAG,CAAC,CAAC,CAAC;YACzC,IAAI,CAAC,KAAK,SAAS,IAAI,CAAC,GAAG,SAAS,EAAE,CAAC;gBACrC,SAAS,GAAG,CAAC,CAAC;gBACd,OAAO,GAAG,CAAC,CAAC;YACd,CAAC;QACH,CAAC;QACD,IAAI,SAAS,GAAG,aAAa;YAAE,SAAS;QAExC,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;QAC7B,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,GAAG,CAAC,CAAW,CAAC;QAC1C,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,GAAG,UAAU,GAAG,CAAC,CAAW,CAAC;QAC7C,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,GAAG,UAAU,GAAG,CAAC,CAAW,CAAC;QAE7C,IAAI,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QACpB,IAAI,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QACpB,IAAI,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QACpB,IAAI,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QAEpB,EAAE,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC,GAAG,KAAK,CAAC;QAC5B,EAAE,GAAG,CAAC,EAAE,GAAG,MAAM,CAAC,GAAG,KAAK,CAAC;QAC3B,EAAE,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC,GAAG,KAAK,CAAC;QAC5B,EAAE,GAAG,CAAC,EAAE,GAAG,MAAM,CAAC,GAAG,KAAK,CAAC;QAE3B,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC;QAC9C,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;QAC/C,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC;QAC9C,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;QAE/C,UAAU,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;IAC7F,CAAC;IAED,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,YAAY,EAAE,CAAC;IAEnD,2EAA2E;IAC3E,yDAAyD;IACzD,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC1D,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IACtD,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IAClD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3C,MAAM,CAAC,GAAG,UAAU,CAAC,CAAC,CAAc,CAAC;QACrC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QACxB,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC5B,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC5B,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC5B,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,UAAU,CAAC;QAC5B,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC;IACzB,CAAC;IACD,MAAM,IAAI,GAAG,UAAU,CAAC,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;IACrE,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,YAAY,EAAE,CAAC;IAE7C,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;IACzD,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IACzB,MAAM,aAAa,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IACxC,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAC1C,MAAM,QAAQ,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,WAAW,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC,CAAW,CAAc,CAAC;QACxD,aAAa,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC;QAC/B,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QACxB,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC5B,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC5B,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC5B,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC;QACxB,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,UAAU,CAAC;IAChC,CAAC;IACD,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;AAC7D,CAAC;AAED,SAAS,YAAY;IACnB,OAAO;QACL,aAAa,EAAE,IAAI,UAAU,CAAC,CAAC,CAAC;QAChC,SAAS,EAAE,IAAI,YAAY,CAAC,CAAC,CAAC;QAC9B,QAAQ,EAAE,IAAI,UAAU,CAAC,CAAC,CAAC;QAC3B,WAAW,EAAE,IAAI,YAAY,CAAC,CAAC,CAAC;KACjC,CAAC;AACJ,CAAC;AAmBD;;;;;;;GAOG;AACH,MAAM,UAAU,UAAU,CACxB,MAAoB,EACpB,UAA6B,EAC7B,OAA0B;IAE1B,MAAM,QAAQ,GAAG,UAAU,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACzE,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;QAC3C,MAAM,IAAI,KAAK,CACb,4CAA4C,QAAQ,mBAAmB,CACxE,CAAC;IACJ,CAAC;IACD,MAAM,UAAU,GAAG,QAAQ,GAAG,CAAC,CAAC;IAEhC,MAAM,OAAO,GAAG,iBAAiB,CAAC,MAAM,EAAE,UAAU,EAAE;QACpD,UAAU;QACV,GAAG,OAAO;KACX,CAAC,CAAC;IAEH,MAAM,OAAO,GAAuB,EAAE,CAAC;IACvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACjD,OAAO,CAAC,IAAI,CAAC;YACX,IAAI,EAAE,IAAI,WAAW,CACnB,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAW,EAClC,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,EACtC,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,EACtC,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CACvC;YACD,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAW;YACtC,UAAU,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC,CAAW;SAC7C,CAAC,CAAC;IACL,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,IAAI,mBAAmB,GAAG,KAAK,CAAC;AAChC,IAAI,0BAA0B,GAAG,KAAK,CAAC;AAEvC;;;GAGG;AACH,MAAM,UAAU,YAAY,CAC1B,MAAoB,EACpB,UAA6B,EAC7B,OAA0B;IAE1B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACzB,mBAAmB,GAAG,IAAI,CAAC;QAC3B,OAAO,CAAC,IAAI,CACV,gFAAgF;YAC9E,qCAAqC,CACxC,CAAC;IACJ,CAAC;IACD,OAAO,UAAU,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;AACjD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,IAAkB,EAClB,IAAuB,EACvB,OAAiC;IAEjC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QAChC,0BAA0B,GAAG,IAAI,CAAC;QAClC,OAAO,CAAC,IAAI,CACV,8FAA8F;YAC5F,qCAAqC,CACxC,CAAC;IACJ,CAAC;IACD,OAAO,iBAAiB,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AAChD,CAAC"}

package/dist/postprocess/segmentation.d.ts

@@ -0,0 +1,61 @@
+/**
+ * Segmentation head postprocessing: YOLO instance-segmentation decoding.
+ *
+ * Compatible with YOLOv8-seg / YOLOv11-seg exports (and any later seg head
+ * sharing the layout). The model produces two output tensors:
+ *
+ * - `output0` of shape `(1, 4 + numClasses + numMaskCoefs, numAnchors)` — the
+ *   same per-anchor predictions as plain YOLO detection plus an extra
+ *   `numMaskCoefs` (typically 32) channels of mask coefficients.
+ * - `output1` of shape `(1, numMaskCoefs, maskH, maskW)` — a set of "prototype"
+ *   masks shared across all anchors.
+ *
+ * The per-anchor decode (xywh→xyxy, undo letterbox, per-class NMS, sort & cap)
+ * is delegated to {@link decodeYoloAnchors}; this module only handles the
+ * mask-specific work: matmul against prototypes, sigmoid, bilinear resize and
+ * thresholding.
+ */
+import { BoundingBox, Mask } from "../types.js";
+export interface DecodeYoloSegOptions {
+    readonly numClasses: number;
+    /** Model input `[width, height]` (post-letterbox). */
+    readonly inputWidth: number;
+    readonly inputHeight: number;
+    /** Original image `[width, height]`. */
+    readonly originalWidth: number;
+    readonly originalHeight: number;
+    /** Letterbox horizontal padding in input-tensor pixels. */
+    readonly padLeft: number;
+    /** Letterbox vertical padding in input-tensor pixels. */
+    readonly padTop: number;
+    /** Letterbox scale factor. */
+    readonly scale: number;
+    readonly confThreshold: number;
+    readonly iouThreshold: number;
+    readonly maxDetections: number;
+    /** Probability cutoff applied to the soft mask. Defaults to `0.5`. */
+    readonly maskThreshold?: number;
+}
+export interface DecodedSegmentation {
+    readonly bbox: BoundingBox;
+    readonly classId: number;
+    readonly confidence: number;
+    /** Binary mask cropped to `bbox`. Width/height match `bbox.asIntXyxy()` extents. */
+    readonly mask: Mask;
+}
+/**
+ * Decode YOLO segmentation raw outputs into a list of segmented instances.
+ *
+ * Compatible with YOLOv8-seg / YOLOv11-seg.
+ *
+ * @param perAnchorData Flat `output0`, length `(4 + numClasses + numMaskCoefs) * numAnchors`.
+ * @param perAnchorDims Dims as reported by ORT, e.g. `[1, 116, 8400]`.
+ * @param prototypeData Flat `output1`, length `numMaskCoefs * maskH * maskW`.
+ * @param prototypeDims Dims as reported by ORT, e.g. `[1, 32, 160, 160]`.
+ */
+export declare function decodeYoloSeg(perAnchorData: Float32Array, perAnchorDims: readonly number[], prototypeData: Float32Array, prototypeDims: readonly number[], options: DecodeYoloSegOptions): DecodedSegmentation[];
+/** @deprecated since 0.2.0 — use {@link decodeYoloSeg}. Will be removed in 0.3.0. */
+export declare function decodeYoloV8Seg(perAnchorData: Float32Array, perAnchorDims: readonly number[], prototypeData: Float32Array, prototypeDims: readonly number[], options: DecodeYoloSegOptions): DecodedSegmentation[];
+/** @deprecated since 0.2.0 — use {@link DecodeYoloSegOptions}. */
+export type DecodeYoloV8SegOptions = DecodeYoloSegOptions;
+//# sourceMappingURL=segmentation.d.ts.map

package/dist/postprocess/segmentation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"segmentation.d.ts","sourceRoot":"","sources":["../../src/postprocess/segmentation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAGH,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAEhD,MAAM,WAAW,oBAAoB;IACnC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,sDAAsD;IACtD,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,wCAAwC;IACxC,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,cAAc,EAAE,MAAM,CAAC;IAChC,2DAA2D;IAC3D,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,yDAAyD;IACzD,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,8BAA8B;IAC9B,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,sEAAsE;IACtE,QAAQ,CAAC,aAAa,CAAC,EAAE,MAAM,CAAC;CACjC;AAED,MAAM,WAAW,mBAAmB;IAClC,QAAQ,CAAC,IAAI,EAAE,WAAW,CAAC;IAC3B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,oFAAoF;IACpF,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC;CACrB;AAED;;;;;;;;;GASG;AACH,wBAAgB,aAAa,CAC3B,aAAa,EAAE,YAAY,EAC3B,aAAa,EAAE,SAAS,MAAM,EAAE,EAChC,aAAa,EAAE,YAAY,EAC3B,aAAa,EAAE,SAAS,MAAM,EAAE,EAChC,OAAO,EAAE,oBAAoB,GAC5B,mBAAmB,EAAE,CA+HvB;AAYD,qFAAqF;AACrF,wBAAgB,eAAe,CAC7B,aAAa,EAAE,YAAY,EAC3B,aAAa,EAAE,SAAS,MAAM,EAAE,EAChC,aAAa,EAAE,YAAY,EAC3B,aAAa,EAAE,SAAS,MAAM,EAAE,EAChC,OAAO,EAAE,oBAAoB,GAC5B,mBAAmB,EAAE,CAevB;AAED,kEAAkE;AAClE,MAAM,MAAM,sBAAsB,GAAG,oBAAoB,CAAC"}

package/dist/postprocess/segmentation.js

@@ -0,0 +1,184 @@
+/**
+ * Segmentation head postprocessing: YOLO instance-segmentation decoding.
+ *
+ * Compatible with YOLOv8-seg / YOLOv11-seg exports (and any later seg head
+ * sharing the layout). The model produces two output tensors:
+ *
+ * - `output0` of shape `(1, 4 + numClasses + numMaskCoefs, numAnchors)` — the
+ *   same per-anchor predictions as plain YOLO detection plus an extra
+ *   `numMaskCoefs` (typically 32) channels of mask coefficients.
+ * - `output1` of shape `(1, numMaskCoefs, maskH, maskW)` — a set of "prototype"
+ *   masks shared across all anchors.
+ *
+ * The per-anchor decode (xywh→xyxy, undo letterbox, per-class NMS, sort & cap)
+ * is delegated to {@link decodeYoloAnchors}; this module only handles the
+ * mask-specific work: matmul against prototypes, sigmoid, bilinear resize and
+ * thresholding.
+ */
+import { decodeYoloAnchors } from "./detection.js";
+import { BoundingBox, Mask } from "../types.js";
+/**
+ * Decode YOLO segmentation raw outputs into a list of segmented instances.
+ *
+ * Compatible with YOLOv8-seg / YOLOv11-seg.
+ *
+ * @param perAnchorData Flat `output0`, length `(4 + numClasses + numMaskCoefs) * numAnchors`.
+ * @param perAnchorDims Dims as reported by ORT, e.g. `[1, 116, 8400]`.
+ * @param prototypeData Flat `output1`, length `numMaskCoefs * maskH * maskW`.
+ * @param prototypeDims Dims as reported by ORT, e.g. `[1, 32, 160, 160]`.
+ */
+export function decodeYoloSeg(perAnchorData, perAnchorDims, prototypeData, prototypeDims, options) {
+    // Strip batch dim from prototypes; perAnchor batch is validated by the helper.
+    let pDims = prototypeDims;
+    if (pDims.length === 4) {
+        if (pDims[0] !== 1) {
+            throw new Error(`decodeYoloSeg: expected batch size 1 in prototypes, got ${pDims[0]}.`);
+        }
+        pDims = [pDims[1], pDims[2], pDims[3]];
+    }
+    if (pDims.length !== 3) {
+        throw new Error(`decodeYoloSeg: expected 3-D prototypes after batch removal, got dims=${JSON.stringify(prototypeDims)}.`);
+    }
+    const numMaskCoefs = pDims[0];
+    const maskH = pDims[1];
+    const maskW = pDims[2];
+    const channels = perAnchorDims.length === 3 ? perAnchorDims[1] : perAnchorDims[0];
+    const numAnchors = perAnchorDims.length === 3 ? perAnchorDims[2] : perAnchorDims[1];
+    if (channels === undefined || numAnchors === undefined) {
+        throw new Error(`decodeYoloSeg: cannot read channels/numAnchors from dims=${JSON.stringify(perAnchorDims)}.`);
+    }
+    const expectedChannels = 4 + options.numClasses + numMaskCoefs;
+    if (channels !== expectedChannels) {
+        throw new Error(`decodeYoloSeg: channels=${channels} does not match 4 + numClasses(${options.numClasses}) + numMaskCoefs(${numMaskCoefs}) = ${expectedChannels}.`);
+    }
+    if (prototypeData.length !== numMaskCoefs * maskH * maskW) {
+        throw new Error(`decodeYoloSeg: prototype length ${prototypeData.length} does not match dims=${JSON.stringify(prototypeDims)}.`);
+    }
+    const decoded = decodeYoloAnchors(perAnchorData, perAnchorDims, {
+        numClasses: options.numClasses,
+        originalWidth: options.originalWidth,
+        originalHeight: options.originalHeight,
+        padLeft: options.padLeft,
+        padTop: options.padTop,
+        scale: options.scale,
+        confThreshold: options.confThreshold,
+        iouThreshold: options.iouThreshold,
+        maxDetections: options.maxDetections,
+    });
+    if (decoded.anchorIndices.length === 0)
+        return [];
+    const maskThreshold = options.maskThreshold ?? 0.5;
+    const scaleX = maskW / options.inputWidth;
+    const scaleY = maskH / options.inputHeight;
+    const protoPlane = maskH * maskW;
+    const coefBase = (4 + options.numClasses) * numAnchors;
+    const results = [];
+    for (let i = 0; i < decoded.anchorIndices.length; i++) {
+        const a = decoded.anchorIndices[i];
+        const x1 = decoded.boxesXyxy[i * 4];
+        const y1 = decoded.boxesXyxy[i * 4 + 1];
+        const x2 = decoded.boxesXyxy[i * 4 + 2];
+        const y2 = decoded.boxesXyxy[i * 4 + 3];
+        const bbox = new BoundingBox(x1, y1, x2, y2);
+        const classId = decoded.classIds[i];
+        const confidence = decoded.confidences[i];
+        const bboxW = Math.max(0, Math.trunc(x2) - Math.trunc(x1));
+        const bboxH = Math.max(0, Math.trunc(y2) - Math.trunc(y1));
+        if (bboxW === 0 || bboxH === 0) {
+            results.push({ bbox, classId, confidence, mask: new Mask(new Uint8Array(0), 0, 0) });
+            continue;
+        }
+        // Bbox in input-tensor coords, then in low-res mask coords.
+        const ibx1 = x1 * options.scale + options.padLeft;
+        const iby1 = y1 * options.scale + options.padTop;
+        const ibx2 = x2 * options.scale + options.padLeft;
+        const iby2 = y2 * options.scale + options.padTop;
+        const mbx1 = Math.max(0, Math.floor(ibx1 * scaleX));
+        const mby1 = Math.max(0, Math.floor(iby1 * scaleY));
+        const mbx2 = Math.min(maskW, Math.ceil(ibx2 * scaleX));
+        const mby2 = Math.min(maskH, Math.ceil(iby2 * scaleY));
+        if (mbx2 <= mbx1 || mby2 <= mby1) {
+            results.push({
+                bbox,
+                classId,
+                confidence,
+                mask: new Mask(new Uint8Array(bboxW * bboxH), bboxW, bboxH),
+            });
+            continue;
+        }
+        // Compute soft mask only within the prototype region under this bbox.
+        const cropW = mbx2 - mbx1;
+        const cropH = mby2 - mby1;
+        const softCrop = new Float32Array(cropW * cropH);
+        for (let y = 0; y < cropH; y++) {
+            const py = mby1 + y;
+            for (let x = 0; x < cropW; x++) {
+                const px = mbx1 + x;
+                let sum = 0;
+                for (let kk = 0; kk < numMaskCoefs; kk++) {
+                    const coef = perAnchorData[coefBase + kk * numAnchors + a];
+                    const proto = prototypeData[kk * protoPlane + py * maskW + px];
+                    sum += coef * proto;
+                }
+                softCrop[y * cropW + x] = sigmoid(sum);
+            }
+        }
+        const resized = resizeBilinear(softCrop, cropW, cropH, bboxW, bboxH);
+        const binary = new Uint8Array(bboxW * bboxH);
+        for (let j = 0; j < binary.length; j++) {
+            binary[j] = resized[j] >= maskThreshold ? 255 : 0;
+        }
+        results.push({ bbox, classId, confidence, mask: new Mask(binary, bboxW, bboxH) });
+    }
+    return results;
+}
+function sigmoid(x) {
+    if (x >= 0) {
+        return 1 / (1 + Math.exp(-x));
+    }
+    const e = Math.exp(x);
+    return e / (1 + e);
+}
+let _warnedDecodeYoloV8Seg = false;
+/** @deprecated since 0.2.0 — use {@link decodeYoloSeg}. Will be removed in 0.3.0. */
+export function decodeYoloV8Seg(perAnchorData, perAnchorDims, prototypeData, prototypeDims, options) {
+    if (!_warnedDecodeYoloV8Seg) {
+        _warnedDecodeYoloV8Seg = true;
+        console.warn("[@ort-vision-sdk/web] decodeYoloV8Seg is deprecated since 0.2.0; use decodeYoloSeg. " +
+            "The alias will be removed in 0.3.0.");
+    }
+    return decodeYoloSeg(perAnchorData, perAnchorDims, prototypeData, prototypeDims, options);
+}
+function resizeBilinear(src, srcWidth, srcHeight, targetWidth, targetHeight) {
+    const out = new Float32Array(targetWidth * targetHeight);
+    if (targetWidth === 0 || targetHeight === 0 || srcWidth === 0 || srcHeight === 0) {
+        return out;
+    }
+    if (targetWidth === srcWidth && targetHeight === srcHeight) {
+        out.set(src);
+        return out;
+    }
+    const sx = srcWidth / targetWidth;
+    const sy = srcHeight / targetHeight;
+    for (let y = 0; y < targetHeight; y++) {
+        const yy = (y + 0.5) * sy - 0.5;
+        const y0 = Math.max(0, Math.floor(yy));
+        const y1 = Math.min(srcHeight - 1, y0 + 1);
+        const wy = Math.max(0, Math.min(1, yy - y0));
+        for (let x = 0; x < targetWidth; x++) {
+            const xx = (x + 0.5) * sx - 0.5;
+            const x0 = Math.max(0, Math.floor(xx));
+            const x1 = Math.min(srcWidth - 1, x0 + 1);
+            const wx = Math.max(0, Math.min(1, xx - x0));
+            const v00 = src[y0 * srcWidth + x0];
+            const v01 = src[y0 * srcWidth + x1];
+            const v10 = src[y1 * srcWidth + x0];
+            const v11 = src[y1 * srcWidth + x1];
+            const top = v00 * (1 - wx) + v01 * wx;
+            const bot = v10 * (1 - wx) + v11 * wx;
+            out[y * targetWidth + x] = top * (1 - wy) + bot * wy;
+        }
+    }
+    return out;
+}
+//# sourceMappingURL=segmentation.js.map

package/dist/postprocess/segmentation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"segmentation.js","sourceRoot":"","sources":["../../src/postprocess/segmentation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AA+BhD;;;;;;;;;GASG;AACH,MAAM,UAAU,aAAa,CAC3B,aAA2B,EAC3B,aAAgC,EAChC,aAA2B,EAC3B,aAAgC,EAChC,OAA6B;IAE7B,+EAA+E;IAC/E,IAAI,KAAK,GAAG,aAAa,CAAC;IAC1B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,2DAA2D,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAC1F,CAAC;QACD,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC,CAAW,EAAE,KAAK,CAAC,CAAC,CAAW,EAAE,KAAK,CAAC,CAAC,CAAW,CAAC,CAAC;IACvE,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CACb,wEAAwE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,CACzG,CAAC;IACJ,CAAC;IACD,MAAM,YAAY,GAAG,KAAK,CAAC,CAAC,CAAW,CAAC;IACxC,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAW,CAAC;IACjC,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAW,CAAC;IAEjC,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;IAClF,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;IACpF,IAAI,QAAQ,KAAK,SAAS,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QACvD,MAAM,IAAI,KAAK,CACb,4DAA4D,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,CAC7F,CAAC;IACJ,CAAC;IAED,MAAM,gBAAgB,GAAG,CAAC,GAAG,OAAO,CAAC,UAAU,GAAG,YAAY,CAAC;IAC/D,IAAI,QAAQ,KAAK,gBAAgB,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CACb,2BAA2B,QAAQ,kCAAkC,OAAO,CAAC,UAAU,oBAAoB,YAAY,OAAO,gBAAgB,GAAG,CAClJ,CAAC;IACJ,CAAC;IACD,IAAI,aAAa,CAAC,MAAM,KAAK,YAAY,GAAG,KAAK,GAAG,KAAK,EAAE,CAAC;QAC1D,MAAM,IAAI,KAAK,CACb,mCAAmC,aAAa,CAAC,MAAM,wBAAwB,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,CAChH,CAAC;IACJ,CAAC;IAED,MAAM,OAAO,GAAG,iBAAiB,CAAC,aAAa,EAAE,aAAa,EAAE;QAC9D,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,aAAa,EAAE,OAAO,CAAC,aAAa;QACpC,cAAc,EAAE,OAAO,CAAC,cAAc;QACtC,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,aAAa,EAAE,OAAO,CAAC,aAAa;QACpC,YAAY,EAAE,OAAO,CAAC,YAAY;QAClC,aAAa,EAAE,OAAO,CAAC,aAAa;KACrC,CAAC,CAAC;IAEH,IAAI,OAAO,CAAC,aAAa,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAElD,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,GAAG,CAAC;IACnD,MAAM,MAAM,GAAG,KAAK,GAAG,OAAO,CAAC,UAAU,CAAC;IAC1C,MAAM,MAAM,GAAG,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC;IAC3C,MAAM,UAAU,GAAG,KAAK,GAAG,KAAK,CAAC;IACjC,MAAM,QAAQ,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,UAAU,CAAC,GAAG,UAAU,CAAC;IAEvD,MAAM,OAAO,GAA0B,EAAE,CAAC;IAE1C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtD,MAAM,CAAC,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAW,CAAC;QAC7C,MAAM,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAW,CAAC;QAC9C,MAAM,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;QAClD,MAAM,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;QAClD,MAAM,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAW,CAAC;QAClD,MAAM,IAAI,GAAG,IAAI,WAAW,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QAC7C,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAW,CAAC;QAC9C,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAW,CAAC;QAEpD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;QAC3D,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;QAE3D,IAAI,KAAK,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,IAAI,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YACrF,SAAS;QACX,CAAC;QAED,4DAA4D;QAC5D,MAAM,IAAI,GAAG,EAAE,GAAG,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC;QAClD,MAAM,IAAI,GAAG,EAAE,GAAG,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC;QACjD,MAAM,IAAI,GAAG,EAAE,GAAG,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC;QAClD,MAAM,IAAI,GAAG,EAAE,GAAG,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC;QACpD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC;QACpD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC;QACvD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC;QAEvD,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;YACjC,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI;gBACJ,OAAO;gBACP,UAAU;gBACV,IAAI,EAAE,IAAI,IAAI,CAAC,IAAI,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC;aAC5D,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,sEAAsE;QACtE,MAAM,KAAK,GAAG,IAAI,GAAG,IAAI,CAAC;QAC1B,MAAM,KAAK,GAAG,IAAI,GAAG,IAAI,CAAC;QAC1B,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC;QACjD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,MAAM,EAAE,GAAG,IAAI,GAAG,CAAC,CAAC;YACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC/B,MAAM,EAAE,GAAG,IAAI,GAAG,CAAC,CAAC;gBACpB,IAAI,GAAG,GAAG,CAAC,CAAC;gBACZ,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,YAAY,EAAE,EAAE,EAAE,EAAE,CAAC;oBACzC,MAAM,IAAI,GAAG,aAAa,CAAC,QAAQ,GAAG,EAAE,GAAG,UAAU,GAAG,CAAC,CAAC,CAAC;oBAC3D,MAAM,KAAK,GAAG,aAAa,CAAC,EAAE,GAAG,UAAU,GAAG,EAAE,GAAG,KAAK,GAAG,EAAE,CAAC,CAAC;oBAC/D,GAAG,IAAI,IAAI,GAAG,KAAK,CAAC;gBACtB,CAAC;gBACD,QAAQ,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAG,cAAc,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QACrE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC;QAC7C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,MAAM,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,aAAa,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC;QAED,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;IACpF,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,OAAO,CAAC,CAAS;IACxB,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACX,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC;IACD,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IACtB,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;AACrB,CAAC;AAED,IAAI,sBAAsB,GAAG,KAAK,CAAC;AAEnC,qFAAqF;AACrF,MAAM,UAAU,eAAe,CAC7B,aAA2B,EAC3B,aAAgC,EAChC,aAA2B,EAC3B,aAAgC,EAChC,OAA6B;IAE7B,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC5B,sBAAsB,GAAG,IAAI,CAAC;QAC9B,OAAO,CAAC,IAAI,CACV,sFAAsF;YACpF,qCAAqC,CACxC,CAAC;IACJ,CAAC;IACD,OAAO,aAAa,CAClB,aAAa,EACb,aAAa,EACb,aAAa,EACb,aAAa,EACb,OAAO,CACR,CAAC;AACJ,CAAC;AAKD,SAAS,cAAc,CACrB,GAAiB,EACjB,QAAgB,EAChB,SAAiB,EACjB,WAAmB,EACnB,YAAoB;IAEpB,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,WAAW,GAAG,YAAY,CAAC,CAAC;IACzD,IAAI,WAAW,KAAK,CAAC,IAAI,YAAY,KAAK,CAAC,IAAI,QAAQ,KAAK,CAAC,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;QACjF,OAAO,GAAG,CAAC;IACb,CAAC;IACD,IAAI,WAAW,KAAK,QAAQ,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;QAC3D,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACb,OAAO,GAAG,CAAC;IACb,CAAC;IACD,MAAM,EAAE,GAAG,QAAQ,GAAG,WAAW,CAAC;IAClC,MAAM,EAAE,GAAG,SAAS,GAAG,YAAY,CAAC;IACpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC;QAChC,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;QACvC,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;QAC3C,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAC7C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC;YAChC,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;YACvC,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;YAC1C,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;YAC7C,MAAM,GAAG,GAAG,GAAG,CAAC,EAAE,GAAG,QAAQ,GAAG,EAAE,CAAC,CAAC;YACpC,MAAM,GAAG,GAAG,GAAG,CAAC,EAAE,GAAG,QAAQ,GAAG,EAAE,CAAC,CAAC;YACpC,MAAM,GAAG,GAAG,GAAG,CAAC,EAAE,GAAG,QAAQ,GAAG,EAAE,CAAC,CAAC;YACpC,MAAM,GAAG,GAAG,GAAG,CAAC,EAAE,GAAG,QAAQ,GAAG,EAAE,CAAC,CAAC;YACpC,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC;YACtC,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC;YACtC,GAAG,CAAC,CAAC,GAAG,WAAW,GAAG,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC;QACvD,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC"}

package/dist/preprocess/image.d.ts

@@ -0,0 +1,71 @@
+/**
+ * Composable image preprocessing primitives (resize, normalize, letterbox, layout).
+ *
+ * Mirrors the Python `preprocess.image` module. Operates on the canonical
+ * {@link RGBImage} (HWC RGB uint8) and produces uint8 or float32 buffers
+ * depending on the operation.
+ */
+import type * as ort from "onnxruntime-web";
+import { RGBImage } from "../types.js";
+/** Resize an image to `(targetWidth, targetHeight)` using high-quality canvas resampling. */
+export declare function resize(image: RGBImage, targetWidth: number, targetHeight: number): RGBImage;
+/**
+ * Convert a uint8 image to a normalized float32 array (HWC layout preserved).
+ *
+ * Applies `(pixel * scale - mean) / std` channel-wise.
+ */
+export declare function normalize(image: RGBImage, mean: readonly [number, number, number], std: readonly [number, number, number], scale?: number): Float32Array;
+/** Convert a uint8 image to a `Float32Array` in `[0, 1]` (HWC layout preserved). */
+export declare function toFloat32(image: RGBImage, scale?: number): Float32Array;
+/**
+ * Transpose interleaved HWC data to planar CHW layout.
+ *
+ * @param hwc Source array of length `width * height * channels`.
+ */
+export declare function toCHW(hwc: Float32Array, width: number, height: number, channels?: number): Float32Array;
+/** Wrap a Float32 buffer into an `ort.Tensor`. */
+export declare function toFloat32Tensor(data: Float32Array, dims: readonly number[]): ort.Tensor;
+/**
+ * Convert an HWC uint8 image to a CHW `Float32Array` scaled to `[0, 1]`.
+ *
+ * Mirrors `torchvision.transforms.ToTensor()` semantics: HWC → CHW,
+ * `uint8 → float32 / 255`. Useful as input to YOLO-style detectors that
+ * don't require ImageNet normalization.
+ *
+ * @returns CHW `Float32Array` of length `width * height * 3`.
+ */
+export declare function toTensor(image: RGBImage): Float32Array;
+/**
+ * Convert an HWC BGR uint8 buffer (OpenCV layout) to the SDK's HWC RGB.
+ *
+ * Use when you receive image bytes from `cv2.imencode` over the wire and
+ * want to feed them to the SDK without going through a canvas decode.
+ *
+ * @param bgr Flat BGR Uint8Array of length `width * height * 3`.
+ */
+export declare function fromCv2(bgr: Uint8Array, width: number, height: number): RGBImage;
+/**
+ * Convert the SDK's HWC RGB image to an HWC BGR `Uint8Array` (OpenCV layout).
+ *
+ * Useful for round-tripping data to a Python OpenCV consumer.
+ */
+export declare function toCv2(image: RGBImage): Uint8Array;
+export interface LetterboxResult {
+    /** The padded image at the target size. */
+    readonly image: RGBImage;
+    /** The factor applied to the original image (`< 1` if downscaled). */
+    readonly scale: number;
+    /** Horizontal padding in pixels (left side; right side has the same or +1). */
+    readonly padLeft: number;
+    /** Vertical padding in pixels (top side). */
+    readonly padTop: number;
+}
+/**
+ * Resize preserving aspect ratio, padding to `(targetWidth, targetHeight)`
+ * with a constant fill color.
+ *
+ * Standard YOLO preprocessing — returning `scale` and `padLeft`/`padTop`
+ * lets callers map detections back to the original image coordinates.
+ */
+export declare function letterbox(image: RGBImage, targetWidth: number, targetHeight: number, fill?: readonly [number, number, number]): LetterboxResult;
+//# sourceMappingURL=image.d.ts.map

package/dist/preprocess/image.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"image.d.ts","sourceRoot":"","sources":["../../src/preprocess/image.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAGH,OAAO,KAAK,KAAK,GAAG,MAAM,iBAAiB,CAAC;AAQ5C,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAEvC,6FAA6F;AAC7F,wBAAgB,MAAM,CACpB,KAAK,EAAE,QAAQ,EACf,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,GACnB,QAAQ,CAoBV;AAED;;;;GAIG;AACH,wBAAgB,SAAS,CACvB,KAAK,EAAE,QAAQ,EACf,IAAI,EAAE,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,EACvC,GAAG,EAAE,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,EACtC,KAAK,GAAE,MAAgB,GACtB,YAAY,CAed;AAED,oFAAoF;AACpF,wBAAgB,SAAS,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,GAAE,MAAgB,GAAG,YAAY,CAOhF;AAED;;;;GAIG;AACH,wBAAgB,KAAK,CACnB,GAAG,EAAE,YAAY,EACjB,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,QAAQ,GAAE,MAAU,GACnB,YAAY,CAmBd;AAED,kDAAkD;AAClD,wBAAgB,eAAe,CAC7B,IAAI,EAAE,YAAY,EAClB,IAAI,EAAE,SAAS,MAAM,EAAE,GACtB,GAAG,CAAC,MAAM,CAEZ;AAED;;;;;;;;GAQG;AACH,wBAAgB,QAAQ,CAAC,KAAK,EAAE,QAAQ,GAAG,YAAY,CAGtD;AAED;;;;;;;GAOG;AACH,wBAAgB,OAAO,CAAC,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,QAAQ,CAahF;AAED;;;;GAIG;AACH,wBAAgB,KAAK,CAAC,KAAK,EAAE,QAAQ,GAAG,UAAU,CASjD;AAED,MAAM,WAAW,eAAe;IAC9B,2CAA2C;IAC3C,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC;IACzB,sEAAsE;IACtE,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,+EAA+E;IAC/E,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,6CAA6C;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;CACzB;AAED;;;;;;GAMG;AACH,wBAAgB,SAAS,CACvB,KAAK,EAAE,QAAQ,EACf,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,EACpB,IAAI,GAAE,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAmB,GACxD,eAAe,CA+BjB"}