tempest-react-sdk 0.22.0 → 0.24.0

package/dist/vision/results.js

@@ -0,0 +1,165 @@
+//#region src/vision/results.ts
+var e = class {
+	xyxy;
+	cls;
+	conf;
+	origShape;
+	constructor(e, t, n, r) {
+		this.xyxy = e, this.cls = t, this.conf = n, this.origShape = r;
+	}
+	get length() {
+		return this.cls.length;
+	}
+	get shape() {
+		return [this.length, 4];
+	}
+	get xywh() {
+		let e = new Float32Array(this.xyxy.length);
+		for (let t = 0; t < this.length; t++) {
+			let n = this.xyxy[t * 4], r = this.xyxy[t * 4 + 1], i = this.xyxy[t * 4 + 2], a = this.xyxy[t * 4 + 3];
+			e[t * 4] = (n + i) / 2, e[t * 4 + 1] = (r + a) / 2, e[t * 4 + 2] = i - n, e[t * 4 + 3] = a - r;
+		}
+		return e;
+	}
+	get xyxyn() {
+		let [e, t] = this.origShape, n = new Float32Array(this.xyxy.length);
+		if (this.length === 0 || t <= 0 || e <= 0) return n;
+		for (let r = 0; r < this.length; r++) n[r * 4] = this.xyxy[r * 4] / t, n[r * 4 + 1] = this.xyxy[r * 4 + 1] / e, n[r * 4 + 2] = this.xyxy[r * 4 + 2] / t, n[r * 4 + 3] = this.xyxy[r * 4 + 3] / e;
+		return n;
+	}
+	get xywhn() {
+		let e = this.xywh, [t, n] = this.origShape;
+		if (this.length === 0 || n <= 0 || t <= 0) return e;
+		for (let r = 0; r < this.length; r++) e[r * 4] = e[r * 4] / n, e[r * 4 + 1] = e[r * 4 + 1] / t, e[r * 4 + 2] = e[r * 4 + 2] / n, e[r * 4 + 3] = e[r * 4 + 3] / t;
+		return e;
+	}
+	get data() {
+		let e = new Float32Array(this.length * 6);
+		for (let t = 0; t < this.length; t++) e[t * 6] = this.xyxy[t * 4], e[t * 6 + 1] = this.xyxy[t * 4 + 1], e[t * 6 + 2] = this.xyxy[t * 4 + 2], e[t * 6 + 3] = this.xyxy[t * 4 + 3], e[t * 6 + 4] = this.conf[t], e[t * 6 + 5] = this.cls[t];
+		return e;
+	}
+}, t = class {
+	data;
+	constructor(e) {
+		this.data = e;
+	}
+	get length() {
+		return this.data.length;
+	}
+	get shape() {
+		return [this.length];
+	}
+	get top1() {
+		if (this.data.length === 0) return 0;
+		let e = 0, t = this.data[0];
+		for (let n = 1; n < this.data.length; n++) {
+			let r = this.data[n];
+			r > t && (e = n, t = r);
+		}
+		return e;
+	}
+	get top1conf() {
+		return this.data.length === 0 ? 0 : this.data[this.top1];
+	}
+	get top5() {
+		return this._topK(5).indices;
+	}
+	get top5conf() {
+		return this._topK(5).values;
+	}
+	_topK(e) {
+		let t = Math.min(e, this.data.length), n = [];
+		for (let e = 0; e < this.data.length; e++) n.push(e);
+		n.sort((e, t) => this.data[t] - this.data[e]);
+		let r = new Int32Array(t), i = new Float32Array(t);
+		for (let e = 0; e < t; e++) r[e] = n[e], i[e] = this.data[n[e]];
+		return {
+			indices: r,
+			values: i
+		};
+	}
+}, n = class {
+	data;
+	xyxy;
+	origShape;
+	constructor(e, t, n) {
+		this.data = e, this.xyxy = t, this.origShape = n;
+	}
+	get length() {
+		return this.data.length;
+	}
+	get shape() {
+		return [this.length];
+	}
+	[Symbol.iterator]() {
+		return this.data[Symbol.iterator]();
+	}
+}, r = class {
+	boxes;
+	detections;
+	names;
+	origImg;
+	origShape;
+	path;
+	speed;
+	constructor(e, t, n, r, i, a = null, o = {}) {
+		this.boxes = e, this.detections = t, this.names = n, this.origImg = r, this.origShape = i, this.path = a, this.speed = o;
+	}
+	get length() {
+		return this.detections.length;
+	}
+	get(e) {
+		return this.detections[e];
+	}
+	[Symbol.iterator]() {
+		return this.detections[Symbol.iterator]();
+	}
+}, i = class {
+	probs;
+	result;
+	names;
+	origImg;
+	origShape;
+	path;
+	speed;
+	constructor(e, t, n, r, i, a = null, o = {}) {
+		this.probs = e, this.result = t, this.names = n, this.origImg = r, this.origShape = i, this.path = a, this.speed = o;
+	}
+	get cls() {
+		return this.probs.top1;
+	}
+	get conf() {
+		return this.probs.top1conf;
+	}
+	get name() {
+		return this.names[this.cls] ?? `class_${this.cls}`;
+	}
+	get probabilities() {
+		return this.result.probabilities;
+	}
+}, a = class {
+	boxes;
+	masks;
+	detections;
+	names;
+	origImg;
+	origShape;
+	path;
+	speed;
+	constructor(e, t, n, r, i, a, o = null, s = {}) {
+		this.boxes = e, this.masks = t, this.detections = n, this.names = r, this.origImg = i, this.origShape = a, this.path = o, this.speed = s;
+	}
+	get length() {
+		return this.detections.length;
+	}
+	get(e) {
+		return this.detections[e];
+	}
+	[Symbol.iterator]() {
+		return this.detections[Symbol.iterator]();
+	}
+};
+//#endregion
+export { e as Boxes, i as ClassificationResults, r as DetectionResults, n as Masks, t as Probs, a as SegmentationResults };
+
+//# sourceMappingURL=results.js.map

package/dist/vision/results.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"results.js","names":[],"sources":["../../src/vision/results.ts"],"sourcesContent":["/**\n * Per-image result envelopes — Ultralytics-style `Results` for the SDK.\n *\n * Each `predict()` call returns a 1-element array of these envelopes,\n * mirroring `YOLO(\"img.jpg\")`. Each envelope holds:\n *\n * - A bulk-array view of the predictions (`Boxes`, `Probs`, `Masks`) with\n *   the exact attribute names Ultralytics uses (`xyxy`, `xywh`, `xyxyn`,\n *   `xywhn`, `cls`, `conf`, `data`, `top1`, `top5`).\n * - Per-instance dataclasses (`DetectionResult`, `SegmentationResult`,\n *   `ClassProbability`) for callers who prefer the OO interface.\n * - `names`: `Record<number, string>` matching Ultralytics' `model.names`.\n * - `origImg` / `origShape` / `path`: provenance for the original input.\n */\n\nimport {\n    type ClassificationResult,\n    type DetectionResult,\n    type RGBImage,\n    type SegmentationResult,\n} from \"./types\";\n\n/**\n * Bulk numpy-style view of detected boxes for a single image.\n *\n * Mirrors Ultralytics' `Boxes` interface. Coordinates in {@link xyxy} and\n * {@link xywh} are absolute pixels in the original image; the `*n` variants\n * are normalized to `[0, 1]` using `origShape`.\n */\nexport class Boxes {\n    /**\n     * @param xyxy Flat array of length `4 * N` in `[x1, y1, x2, y2, ...]` order.\n     * @param cls One class index per box, length `N`.\n     * @param conf One confidence per box, length `N`.\n     * @param origShape `[height, width]` of the original image.\n     */\n    constructor(\n        public readonly xyxy: Float32Array,\n        public readonly cls: Int32Array,\n        public readonly conf: Float32Array,\n        public readonly origShape: readonly [number, number],\n    ) {}\n\n    /** Number of detected boxes. */\n    get length(): number {\n        return this.cls.length;\n    }\n\n    /** `[N, 4]` shape of the `xyxy` view. */\n    get shape(): readonly [number, number] {\n        return [this.length, 4];\n    }\n\n    /** Boxes as `[N, 4]` `[cx, cy, w, h]` flat array in absolute pixels. */\n    get xywh(): Float32Array {\n        const out = new Float32Array(this.xyxy.length);\n        for (let i = 0; i < this.length; i++) {\n            const x1 = this.xyxy[i * 4] as number;\n            const y1 = this.xyxy[i * 4 + 1] as number;\n            const x2 = this.xyxy[i * 4 + 2] as number;\n            const y2 = this.xyxy[i * 4 + 3] as number;\n            out[i * 4] = (x1 + x2) / 2;\n            out[i * 4 + 1] = (y1 + y2) / 2;\n            out[i * 4 + 2] = x2 - x1;\n            out[i * 4 + 3] = y2 - y1;\n        }\n        return out;\n    }\n\n    /** Boxes as `[N, 4]` `[x1, y1, x2, y2]` normalized to `[0, 1]`. */\n    get xyxyn(): Float32Array {\n        const [h, w] = this.origShape;\n        const out = new Float32Array(this.xyxy.length);\n        if (this.length === 0 || w <= 0 || h <= 0) return out;\n        for (let i = 0; i < this.length; i++) {\n            out[i * 4] = (this.xyxy[i * 4] as number) / w;\n            out[i * 4 + 1] = (this.xyxy[i * 4 + 1] as number) / h;\n            out[i * 4 + 2] = (this.xyxy[i * 4 + 2] as number) / w;\n            out[i * 4 + 3] = (this.xyxy[i * 4 + 3] as number) / h;\n        }\n        return out;\n    }\n\n    /** Boxes as `[N, 4]` `[cx, cy, w, h]` normalized to `[0, 1]`. */\n    get xywhn(): Float32Array {\n        const xywh = this.xywh;\n        const [h, w] = this.origShape;\n        if (this.length === 0 || w <= 0 || h <= 0) return xywh;\n        for (let i = 0; i < this.length; i++) {\n            xywh[i * 4] = (xywh[i * 4] as number) / w;\n            xywh[i * 4 + 1] = (xywh[i * 4 + 1] as number) / h;\n            xywh[i * 4 + 2] = (xywh[i * 4 + 2] as number) / w;\n            xywh[i * 4 + 3] = (xywh[i * 4 + 3] as number) / h;\n        }\n        return xywh;\n    }\n\n    /**\n     * Concatenated `[N, 6]` array of `[x1, y1, x2, y2, conf, cls]`.\n     *\n     * Matches Ultralytics' `boxes.data`.\n     */\n    get data(): Float32Array {\n        const out = new Float32Array(this.length * 6);\n        for (let i = 0; i < this.length; i++) {\n            out[i * 6] = this.xyxy[i * 4] as number;\n            out[i * 6 + 1] = this.xyxy[i * 4 + 1] as number;\n            out[i * 6 + 2] = this.xyxy[i * 4 + 2] as number;\n            out[i * 6 + 3] = this.xyxy[i * 4 + 3] as number;\n            out[i * 6 + 4] = this.conf[i] as number;\n            out[i * 6 + 5] = this.cls[i] as number;\n        }\n        return out;\n    }\n}\n\n/**\n * Top-k classification probabilities for a single image.\n *\n * Mirrors Ultralytics' `Probs` interface.\n */\nexport class Probs {\n    /** @param data `[numClasses]` per-class probabilities, indexed by class id. */\n    constructor(public readonly data: Float32Array) {}\n\n    /** Number of classes. */\n    get length(): number {\n        return this.data.length;\n    }\n\n    /** `[numClasses]` shape of the underlying vector. */\n    get shape(): readonly [number] {\n        return [this.length];\n    }\n\n    /** Index of the most probable class. */\n    get top1(): number {\n        if (this.data.length === 0) return 0;\n        let best = 0;\n        let bestVal = this.data[0] as number;\n        for (let i = 1; i < this.data.length; i++) {\n            const v = this.data[i] as number;\n            if (v > bestVal) {\n                best = i;\n                bestVal = v;\n            }\n        }\n        return best;\n    }\n\n    /** Probability of the top-1 class. */\n    get top1conf(): number {\n        if (this.data.length === 0) return 0;\n        return this.data[this.top1] as number;\n    }\n\n    /** Indices of the top-5 most probable classes, descending. */\n    get top5(): Int32Array {\n        return this._topK(5).indices;\n    }\n\n    /** Probabilities of the top-5 classes, descending. */\n    get top5conf(): Float32Array {\n        return this._topK(5).values;\n    }\n\n    private _topK(k: number): { indices: Int32Array; values: Float32Array } {\n        const n = Math.min(k, this.data.length);\n        const order: number[] = [];\n        for (let i = 0; i < this.data.length; i++) order.push(i);\n        order.sort((a, b) => (this.data[b] as number) - (this.data[a] as number));\n        const indices = new Int32Array(n);\n        const values = new Float32Array(n);\n        for (let i = 0; i < n; i++) {\n            indices[i] = order[i] as number;\n            values[i] = this.data[order[i] as number] as number;\n        }\n        return { indices, values };\n    }\n}\n\n/**\n * Per-instance binary masks for a single image.\n *\n * Each mask is cropped to its instance's bounding box. To paint masks onto\n * a full-image canvas, use `xyxy[i]` as the top-left target.\n */\nexport class Masks {\n    /**\n     * @param data Per-instance binary masks (`Mask` objects from `types.ts`).\n     * @param xyxy Flat `[N, 4]` of bounding-box coordinates in original pixels.\n     * @param origShape `[height, width]` of the original image.\n     */\n    constructor(\n        public readonly data: ReadonlyArray<{\n            readonly data: Uint8Array;\n            readonly width: number;\n            readonly height: number;\n        }>,\n        public readonly xyxy: Float32Array,\n        public readonly origShape: readonly [number, number],\n    ) {}\n\n    /** Number of instance masks. */\n    get length(): number {\n        return this.data.length;\n    }\n\n    /** `[N]` shape of the masks collection. */\n    get shape(): readonly [number] {\n        return [this.length];\n    }\n\n    [Symbol.iterator](): Iterator<{\n        readonly data: Uint8Array;\n        readonly width: number;\n        readonly height: number;\n    }> {\n        return this.data[Symbol.iterator]();\n    }\n}\n\n/**\n * Per-image detection envelope (Ultralytics-style `Results`).\n *\n * Iterating yields per-instance {@link DetectionResult} entries, so legacy\n * code that did `for (const d of detector.predict(img))` only needs an\n * extra `[0]` to bridge:\n *\n * ```typescript\n * for (const d of (await detector.predict(img))[0]) {\n *   console.log(d.cls, d.conf, d.box.xyxy);\n * }\n * ```\n *\n * For numpy-style bulk access, use the `boxes` collection.\n */\nexport class DetectionResults implements Iterable<DetectionResult> {\n    constructor(\n        public readonly boxes: Boxes,\n        public readonly detections: readonly DetectionResult[],\n        public readonly names: Readonly<Record<number, string>>,\n        public readonly origImg: RGBImage,\n        public readonly origShape: readonly [number, number],\n        public readonly path: string | null = null,\n        public readonly speed: Readonly<Record<string, number>> = {},\n    ) {}\n\n    /** Number of surviving detections. */\n    get length(): number {\n        return this.detections.length;\n    }\n\n    /** Index into the per-instance detections. */\n    get(index: number): DetectionResult | undefined {\n        return this.detections[index];\n    }\n\n    [Symbol.iterator](): Iterator<DetectionResult> {\n        return this.detections[Symbol.iterator]();\n    }\n}\n\n/**\n * Per-image classification envelope (Ultralytics-style `Results`).\n */\nexport class ClassificationResults {\n    constructor(\n        public readonly probs: Probs,\n        public readonly result: ClassificationResult,\n        public readonly names: Readonly<Record<number, string>>,\n        public readonly origImg: RGBImage,\n        public readonly origShape: readonly [number, number],\n        public readonly path: string | null = null,\n        public readonly speed: Readonly<Record<string, number>> = {},\n    ) {}\n\n    /** Top-1 class index (Ultralytics-style alias). */\n    get cls(): number {\n        return this.probs.top1;\n    }\n\n    /** Top-1 confidence (Ultralytics-style alias). */\n    get conf(): number {\n        return this.probs.top1conf;\n    }\n\n    /** Top-1 class name. */\n    get name(): string {\n        return this.names[this.cls] ?? `class_${this.cls}`;\n    }\n\n    /** Per-class probability list, sorted descending (legacy field). */\n    get probabilities(): readonly ClassificationResult[\"probabilities\"][number][] {\n        return this.result.probabilities;\n    }\n}\n\n/**\n * Per-image instance-segmentation envelope (Ultralytics-style `Results`).\n *\n * Iterating yields per-instance {@link SegmentationResult} entries. `boxes`\n * and `masks` mirror Ultralytics' bulk-array views.\n */\nexport class SegmentationResults implements Iterable<SegmentationResult> {\n    constructor(\n        public readonly boxes: Boxes,\n        public readonly masks: Masks,\n        public readonly detections: readonly SegmentationResult[],\n        public readonly names: Readonly<Record<number, string>>,\n        public readonly origImg: RGBImage,\n        public readonly origShape: readonly [number, number],\n        public readonly path: string | null = null,\n        public readonly speed: Readonly<Record<string, number>> = {},\n    ) {}\n\n    /** Number of surviving instances. */\n    get length(): number {\n        return this.detections.length;\n    }\n\n    /** Index into the per-instance results. */\n    get(index: number): SegmentationResult | undefined {\n        return this.detections[index];\n    }\n\n    [Symbol.iterator](): Iterator<SegmentationResult> {\n        return this.detections[Symbol.iterator]();\n    }\n}\n"],"mappings":";AA6BA,IAAa,IAAb,MAAmB;CAQK;CACA;CACA;CACA;CAJpB,YACI,GACA,GACA,GACA,GACF;EADkB,AAHA,KAAA,OAAA,GACA,KAAA,MAAA,GACA,KAAA,OAAA,GACA,KAAA,YAAA;CACjB;CAGH,IAAI,SAAiB;EACjB,OAAO,KAAK,IAAI;CACpB;CAGA,IAAI,QAAmC;EACnC,OAAO,CAAC,KAAK,QAAQ,CAAC;CAC1B;CAGA,IAAI,OAAqB;EACrB,IAAM,IAAM,IAAI,aAAa,KAAK,KAAK,MAAM;EAC7C,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,QAAQ,KAAK;GAClC,IAAM,IAAK,KAAK,KAAK,IAAI,IACnB,IAAK,KAAK,KAAK,IAAI,IAAI,IACvB,IAAK,KAAK,KAAK,IAAI,IAAI,IACvB,IAAK,KAAK,KAAK,IAAI,IAAI;GAI7B,AAHA,EAAI,IAAI,MAAM,IAAK,KAAM,GACzB,EAAI,IAAI,IAAI,MAAM,IAAK,KAAM,GAC7B,EAAI,IAAI,IAAI,KAAK,IAAK,GACtB,EAAI,IAAI,IAAI,KAAK,IAAK;EAC1B;EACA,OAAO;CACX;CAGA,IAAI,QAAsB;EACtB,IAAM,CAAC,GAAG,KAAK,KAAK,WACd,IAAM,IAAI,aAAa,KAAK,KAAK,MAAM;EAC7C,IAAI,KAAK,WAAW,KAAK,KAAK,KAAK,KAAK,GAAG,OAAO;EAClD,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,QAAQ,KAI7B,AAHA,EAAI,IAAI,KAAM,KAAK,KAAK,IAAI,KAAgB,GAC5C,EAAI,IAAI,IAAI,KAAM,KAAK,KAAK,IAAI,IAAI,KAAgB,GACpD,EAAI,IAAI,IAAI,KAAM,KAAK,KAAK,IAAI,IAAI,KAAgB,GACpD,EAAI,IAAI,IAAI,KAAM,KAAK,KAAK,IAAI,IAAI,KAAgB;EAExD,OAAO;CACX;CAGA,IAAI,QAAsB;EACtB,IAAM,IAAO,KAAK,MACZ,CAAC,GAAG,KAAK,KAAK;EACpB,IAAI,KAAK,WAAW,KAAK,KAAK,KAAK,KAAK,GAAG,OAAO;EAClD,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,QAAQ,KAI7B,AAHA,EAAK,IAAI,KAAM,EAAK,IAAI,KAAgB,GACxC,EAAK,IAAI,IAAI,KAAM,EAAK,IAAI,IAAI,KAAgB,GAChD,EAAK,IAAI,IAAI,KAAM,EAAK,IAAI,IAAI,KAAgB,GAChD,EAAK,IAAI,IAAI,KAAM,EAAK,IAAI,IAAI,KAAgB;EAEpD,OAAO;CACX;CAOA,IAAI,OAAqB;EACrB,IAAM,IAAM,IAAI,aAAa,KAAK,SAAS,CAAC;EAC5C,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,QAAQ,KAM7B,AALA,EAAI,IAAI,KAAK,KAAK,KAAK,IAAI,IAC3B,EAAI,IAAI,IAAI,KAAK,KAAK,KAAK,IAAI,IAAI,IACnC,EAAI,IAAI,IAAI,KAAK,KAAK,KAAK,IAAI,IAAI,IACnC,EAAI,IAAI,IAAI,KAAK,KAAK,KAAK,IAAI,IAAI,IACnC,EAAI,IAAI,IAAI,KAAK,KAAK,KAAK,IAC3B,EAAI,IAAI,IAAI,KAAK,KAAK,IAAI;EAE9B,OAAO;CACX;AACJ,GAOa,IAAb,MAAmB;CAEa;CAA5B,YAAY,GAAoC;EAApB,KAAA,OAAA;CAAqB;CAGjD,IAAI,SAAiB;EACjB,OAAO,KAAK,KAAK;CACrB;CAGA,IAAI,QAA2B;EAC3B,OAAO,CAAC,KAAK,MAAM;CACvB;CAGA,IAAI,OAAe;EACf,IAAI,KAAK,KAAK,WAAW,GAAG,OAAO;EACnC,IAAI,IAAO,GACP,IAAU,KAAK,KAAK;EACxB,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,KAAK,QAAQ,KAAK;GACvC,IAAM,IAAI,KAAK,KAAK;GACpB,AAAI,IAAI,MACJ,IAAO,GACP,IAAU;EAElB;EACA,OAAO;CACX;CAGA,IAAI,WAAmB;EAEnB,OADI,KAAK,KAAK,WAAW,IAAU,IAC5B,KAAK,KAAK,KAAK;CAC1B;CAGA,IAAI,OAAmB;EACnB,OAAO,KAAK,MAAM,CAAC,CAAC,CAAC;CACzB;CAGA,IAAI,WAAyB;EACzB,OAAO,KAAK,MAAM,CAAC,CAAC,CAAC;CACzB;CAEA,MAAc,GAA0D;EACpE,IAAM,IAAI,KAAK,IAAI,GAAG,KAAK,KAAK,MAAM,GAChC,IAAkB,CAAC;EACzB,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,KAAK,QAAQ,KAAK,EAAM,KAAK,CAAC;EACvD,EAAM,MAAM,GAAG,MAAO,KAAK,KAAK,KAAiB,KAAK,KAAK,EAAa;EACxE,IAAM,IAAU,IAAI,WAAW,CAAC,GAC1B,IAAS,IAAI,aAAa,CAAC;EACjC,KAAK,IAAI,IAAI,GAAG,IAAI,GAAG,KAEnB,AADA,EAAQ,KAAK,EAAM,IACnB,EAAO,KAAK,KAAK,KAAK,EAAM;EAEhC,OAAO;GAAE;GAAS;EAAO;CAC7B;AACJ,GAQa,IAAb,MAAmB;CAOK;CAKA;CACA;CAPpB,YACI,GAKA,GACA,GACF;EADkB,AANA,KAAA,OAAA,GAKA,KAAA,OAAA,GACA,KAAA,YAAA;CACjB;CAGH,IAAI,SAAiB;EACjB,OAAO,KAAK,KAAK;CACrB;CAGA,IAAI,QAA2B;EAC3B,OAAO,CAAC,KAAK,MAAM;CACvB;CAEA,CAAC,OAAO,YAIL;EACC,OAAO,KAAK,KAAK,OAAO,SAAS,CAAC;CACtC;AACJ,GAiBa,IAAb,MAAmE;CAE3C;CACA;CACA;CACA;CACA;CACA;CACA;CAPpB,YACI,GACA,GACA,GACA,GACA,GACA,IAAsC,MACtC,IAA0D,CAAC,GAC7D;EADkB,AANA,KAAA,QAAA,GACA,KAAA,aAAA,GACA,KAAA,QAAA,GACA,KAAA,UAAA,GACA,KAAA,YAAA,GACA,KAAA,OAAA,GACA,KAAA,QAAA;CACjB;CAGH,IAAI,SAAiB;EACjB,OAAO,KAAK,WAAW;CAC3B;CAGA,IAAI,GAA4C;EAC5C,OAAO,KAAK,WAAW;CAC3B;CAEA,CAAC,OAAO,YAAuC;EAC3C,OAAO,KAAK,WAAW,OAAO,SAAS,CAAC;CAC5C;AACJ,GAKa,IAAb,MAAmC;CAEX;CACA;CACA;CACA;CACA;CACA;CACA;CAPpB,YACI,GACA,GACA,GACA,GACA,GACA,IAAsC,MACtC,IAA0D,CAAC,GAC7D;EADkB,AANA,KAAA,QAAA,GACA,KAAA,SAAA,GACA,KAAA,QAAA,GACA,KAAA,UAAA,GACA,KAAA,YAAA,GACA,KAAA,OAAA,GACA,KAAA,QAAA;CACjB;CAGH,IAAI,MAAc;EACd,OAAO,KAAK,MAAM;CACtB;CAGA,IAAI,OAAe;EACf,OAAO,KAAK,MAAM;CACtB;CAGA,IAAI,OAAe;EACf,OAAO,KAAK,MAAM,KAAK,QAAQ,SAAS,KAAK;CACjD;CAGA,IAAI,gBAA0E;EAC1E,OAAO,KAAK,OAAO;CACvB;AACJ,GAQa,IAAb,MAAyE;CAEjD;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CARpB,YACI,GACA,GACA,GACA,GACA,GACA,GACA,IAAsC,MACtC,IAA0D,CAAC,GAC7D;EADkB,AAPA,KAAA,QAAA,GACA,KAAA,QAAA,GACA,KAAA,aAAA,GACA,KAAA,QAAA,GACA,KAAA,UAAA,GACA,KAAA,YAAA,GACA,KAAA,OAAA,GACA,KAAA,QAAA;CACjB;CAGH,IAAI,SAAiB;EACjB,OAAO,KAAK,WAAW;CAC3B;CAGA,IAAI,GAA+C;EAC/C,OAAO,KAAK,WAAW;CAC3B;CAEA,CAAC,OAAO,YAA0C;EAC9C,OAAO,KAAK,WAAW,OAAO,SAAS,CAAC;CAC5C;AACJ"}

package/dist/vision/tasks/base.cjs

@@ -0,0 +1,2 @@
+var e=class{_session;constructor(e){this._session=e}get session(){return this._session}};exports.VisionTask=e;
+//# sourceMappingURL=base.cjs.map

package/dist/vision/tasks/base.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.cjs","names":[],"sources":["../../../src/vision/tasks/base.ts"],"sourcesContent":["/**\n * Common foundation for task-oriented vision SDK objects.\n *\n * The base only owns the {@link OrtSession} — label resolution lives in each\n * subclass because how `numClasses` is read from the model differs per task.\n */\n\nimport type { OrtSession } from \"../core/session\";\n\nexport abstract class VisionTask {\n    protected constructor(protected readonly _session: OrtSession) {}\n\n    /** The underlying {@link OrtSession} used to run inference. */\n    get session(): OrtSession {\n        return this._session;\n    }\n}\n"],"mappings":"AASA,IAAsB,EAAtB,KAAiC,CACY,SAAzC,YAAsB,EAAyC,CAAtB,KAAA,SAAA,CAAuB,CAGhE,IAAI,SAAsB,CACtB,OAAO,KAAK,QAChB,CACJ"}

package/dist/vision/tasks/base.js

@@ -0,0 +1,14 @@
+//#region src/vision/tasks/base.ts
+var e = class {
+	_session;
+	constructor(e) {
+		this._session = e;
+	}
+	get session() {
+		return this._session;
+	}
+};
+//#endregion
+export { e as VisionTask };
+
+//# sourceMappingURL=base.js.map

package/dist/vision/tasks/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","names":[],"sources":["../../../src/vision/tasks/base.ts"],"sourcesContent":["/**\n * Common foundation for task-oriented vision SDK objects.\n *\n * The base only owns the {@link OrtSession} — label resolution lives in each\n * subclass because how `numClasses` is read from the model differs per task.\n */\n\nimport type { OrtSession } from \"../core/session\";\n\nexport abstract class VisionTask {\n    protected constructor(protected readonly _session: OrtSession) {}\n\n    /** The underlying {@link OrtSession} used to run inference. */\n    get session(): OrtSession {\n        return this._session;\n    }\n}\n"],"mappings":";AASA,IAAsB,IAAtB,MAAiC;CACY;CAAzC,YAAsB,GAAyC;EAAtB,KAAA,WAAA;CAAuB;CAGhE,IAAI,UAAsB;EACtB,OAAO,KAAK;CAChB;AACJ"}

package/dist/vision/tasks/classifier.cjs

@@ -0,0 +1,2 @@
+const e=require("../results.cjs"),t=require("../labels.cjs"),n=require("../core/session.cjs"),r=require("../io/image.cjs"),i=require("../preprocess/image.cjs"),a=require("../postprocess/classification.cjs"),o=require("./base.cjs");var s=[.485,.456,.406],c=[.229,.224,.225],l=class l extends o.VisionTask{_labels;_names;_inputSize;_mean;_std;_applySoftmax;constructor(e,t,n,r,i,a,o){super(e),this._labels=t,this._names=n,this._inputSize=r,this._mean=i,this._std=a,this._applySoftmax=o}static async create(e,r){let i=await n.OrtSession.create(e,r),a=t.resolveLabels(r.labels,{numClasses:r.numClasses}),o={};for(let e=0;e<a.length;e++)o[e]=a[e];return new l(i,a,o,r.inputSize??[224,224],r.mean??s,r.std??c,r.applySoftmax??!0)}get labels(){return this._labels}get names(){return this._names}get numClasses(){return this._labels.length}async call(e,t={}){return this.predict(e,t)}async predict(t,n={}){let i=typeof t==`string`?t:null,o=await r.loadImage(t),s=this._preprocess(o),c=await this._session.run({[this._session.inputName]:s}),l=this._session.outputNames[0];if(l===void 0)throw Error(`Classifier model has no outputs.`);let u=c[l];if(u===void 0)throw Error(`Classifier model output ${l} missing from run() result.`);let d=this._postprocess(u.data),{indices:f,values:p}=a.topK(d,n.topK??null),m=[];for(let e=0;e<f.length;e++){let t=f[e],n=this._labels[t]??`class_${t}`;m.push({classId:t,className:n,probability:p[e],cls:t,name:n,conf:p[e]})}if(m.length===0)throw Error(`Classifier produced no probabilities (empty output).`);let h=m[0],g={classId:h.classId,className:h.className,confidence:h.probability,cls:h.classId,name:h.className,conf:h.probability,image:o,probabilities:m},_=[o.height,o.width];return[new e.ClassificationResults(new e.Probs(d),g,this._names,o,_,i)]}_preprocess(e){let[t,n]=this._inputSize,r=i.resize(e,t,n);return i.toFloat32Tensor(i.toCHW(i.normalize(r,this._mean,this._std),r.width,r.height,3),[1,3,r.height,r.width])}_postprocess(e){return this._applySoftmax?a.softmax(e):new Float32Array(e)}};exports.Classifier=l;
+//# sourceMappingURL=classifier.cjs.map

package/dist/vision/tasks/classifier.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"classifier.cjs","names":[],"sources":["../../../src/vision/tasks/classifier.ts"],"sourcesContent":["/**\n * Image classification task using ONNX Runtime Web.\n */\n\nimport type * as ort from \"onnxruntime-web\";\n\nimport { type ModelSource, type OrtSessionOptions, OrtSession } from \"../core/session\";\nimport { type ImageInput, loadImage } from \"../io/image\";\nimport { type LabelSpec, resolveLabels } from \"../labels\";\nimport { softmax, topK } from \"../postprocess/classification\";\nimport { normalize, resize, toCHW, toFloat32Tensor } from \"../preprocess/image\";\nimport { ClassificationResults, Probs } from \"../results\";\nimport { VisionTask } from \"./base\";\nimport { type ClassProbability, type ClassificationResult, type RGBImage } from \"../types\";\n\nconst IMAGENET_MEAN: readonly [number, number, number] = [0.485, 0.456, 0.406];\nconst IMAGENET_STD: readonly [number, number, number] = [0.229, 0.224, 0.225];\n\nexport interface ClassifierOptions extends OrtSessionOptions {\n    /** Class label spec — see {@link resolveLabels}. */\n    readonly labels: LabelSpec;\n    /**\n     * Number of classes the model can predict. Required when `labels` is `null`\n     * or when you want to validate that the supplied labels match the model.\n     */\n    readonly numClasses?: number;\n    /** Model input `[width, height]` in pixels. Defaults to `[224, 224]`. */\n    readonly inputSize?: readonly [number, number];\n    /** Per-channel RGB mean used for normalization. Defaults to ImageNet. */\n    readonly mean?: readonly [number, number, number];\n    /** Per-channel RGB standard deviation. Defaults to ImageNet. */\n    readonly std?: readonly [number, number, number];\n    /**\n     * If `true` (default), apply softmax to the raw model output. Set to\n     * `false` for models whose final layer already produces a probability\n     * distribution.\n     */\n    readonly applySoftmax?: boolean;\n}\n\nexport interface ClassifierPredictOptions {\n    /**\n     * If set, the per-class probability list in `results[0].result.probabilities`\n     * is truncated to the top-K entries. The bulk `probs` view always exposes\n     * the full vector.\n     */\n    readonly topK?: number;\n}\n\n/**\n * Image classifier wrapping an ONNX model with ImageNet-style preprocessing.\n *\n * `predict()` returns `Promise<ClassificationResults[]>` (length 1 for a\n * single image), mirroring Ultralytics' API. The envelope exposes a `probs`\n * collection (`top1`, `top1conf`, `top5`, `top5conf`, `data`) plus the\n * legacy per-class probability list with names already resolved.\n *\n * Defaults: 224×224 RGB input, `float32` normalized with ImageNet mean/std,\n * NCHW layout, batch size 1, softmax applied to the raw output.\n *\n * @example\n * ```typescript\n * const clf = await Classifier.create(\"/models/resnet50.onnx\", {\n *   labels: [\"tench\", \"goldfish\", ...]  // 1000 ImageNet labels\n * });\n * const r = (await clf.predict(\"/images/dog.jpg\"))[0];\n * console.log(r.cls, r.conf, r.name);\n * console.log(r.probs.top5, r.probs.top5conf);\n * ```\n */\nexport class Classifier extends VisionTask {\n    private constructor(\n        session: OrtSession,\n        private readonly _labels: readonly string[],\n        private readonly _names: Readonly<Record<number, string>>,\n        private readonly _inputSize: readonly [number, number],\n        private readonly _mean: readonly [number, number, number],\n        private readonly _std: readonly [number, number, number],\n        private readonly _applySoftmax: boolean,\n    ) {\n        super(session);\n    }\n\n    /** Load the model and resolve labels. */\n    static async create(model: ModelSource, options: ClassifierOptions): Promise<Classifier> {\n        const session = await OrtSession.create(model, options);\n        const labels = resolveLabels(options.labels, { numClasses: options.numClasses });\n        const names: Record<number, string> = {};\n        for (let i = 0; i < labels.length; i++) {\n            names[i] = labels[i] as string;\n        }\n        return new Classifier(\n            session,\n            labels,\n            names,\n            options.inputSize ?? [224, 224],\n            options.mean ?? IMAGENET_MEAN,\n            options.std ?? IMAGENET_STD,\n            options.applySoftmax ?? true,\n        );\n    }\n\n    /** Class labels indexed by class id. */\n    get labels(): readonly string[] {\n        return this._labels;\n    }\n\n    /** Class id → class name dict (matches Ultralytics' `model.names`). */\n    get names(): Readonly<Record<number, string>> {\n        return this._names;\n    }\n\n    /** Number of classes the model can predict. */\n    get numClasses(): number {\n        return this._labels.length;\n    }\n\n    /** Alias for {@link predict} (parity with PyTorch `nn.Module.__call__`). */\n    async call(\n        image: ImageInput,\n        options: ClassifierPredictOptions = {},\n    ): Promise<ClassificationResults[]> {\n        return this.predict(image, options);\n    }\n\n    /** Run classification on a single image. */\n    async predict(\n        image: ImageInput,\n        options: ClassifierPredictOptions = {},\n    ): Promise<ClassificationResults[]> {\n        const path = typeof image === \"string\" ? image : null;\n        const original = await loadImage(image);\n        const tensor = this._preprocess(original);\n        const outputs = await this._session.run({ [this._session.inputName]: tensor });\n        const firstOutputName = this._session.outputNames[0];\n        if (firstOutputName === undefined) {\n            throw new Error(\"Classifier model has no outputs.\");\n        }\n        const raw = outputs[firstOutputName];\n        if (raw === undefined) {\n            throw new Error(\n                `Classifier model output ${firstOutputName} missing from run() result.`,\n            );\n        }\n        const fullProbs = this._postprocess(raw.data as Float32Array);\n\n        const { indices, values } = topK(fullProbs, options.topK ?? null);\n        const probabilities: ClassProbability[] = [];\n        for (let i = 0; i < indices.length; i++) {\n            const id = indices[i] as number;\n            const className = this._labels[id] ?? `class_${id}`;\n            probabilities.push({\n                classId: id,\n                className,\n                probability: values[i] as number,\n                cls: id,\n                name: className,\n                conf: values[i] as number,\n            });\n        }\n        if (probabilities.length === 0) {\n            throw new Error(\"Classifier produced no probabilities (empty output).\");\n        }\n\n        const top = probabilities[0] as ClassProbability;\n        const result: ClassificationResult = {\n            classId: top.classId,\n            className: top.className,\n            confidence: top.probability,\n            cls: top.classId,\n            name: top.className,\n            conf: top.probability,\n            image: original,\n            probabilities,\n        };\n\n        const orig: readonly [number, number] = [original.height, original.width];\n        return [\n            new ClassificationResults(\n                new Probs(fullProbs),\n                result,\n                this._names,\n                original,\n                orig,\n                path,\n            ),\n        ];\n    }\n\n    private _preprocess(image: RGBImage): ort.Tensor {\n        const [tw, th] = this._inputSize;\n        const resized = resize(image, tw, th);\n        const normalized = normalize(resized, this._mean, this._std);\n        const chw = toCHW(normalized, resized.width, resized.height, 3);\n        return toFloat32Tensor(chw, [1, 3, resized.height, resized.width]);\n    }\n\n    private _postprocess(raw: Float32Array): Float32Array {\n        return this._applySoftmax ? softmax(raw) : new Float32Array(raw);\n    }\n}\n"],"mappings":"uOAeA,IAAM,EAAmD,CAAC,KAAO,KAAO,IAAK,EACvE,EAAkD,CAAC,KAAO,KAAO,IAAK,EAsD/D,EAAb,MAAa,UAAmB,EAAA,UAAW,CAGlB,QACA,OACA,WACA,MACA,KACA,cAPrB,YACI,EACA,EACA,EACA,EACA,EACA,EACA,EACF,CACE,MAAM,CAAO,EAPI,KAAA,QAAA,EACA,KAAA,OAAA,EACA,KAAA,WAAA,EACA,KAAA,MAAA,EACA,KAAA,KAAA,EACA,KAAA,cAAA,CAGrB,CAGA,aAAa,OAAO,EAAoB,EAAiD,CACrF,IAAM,EAAU,MAAM,EAAA,WAAW,OAAO,EAAO,CAAO,EAChD,EAAS,EAAA,cAAc,EAAQ,OAAQ,CAAE,WAAY,EAAQ,UAAW,CAAC,EACzE,EAAgC,CAAC,EACvC,IAAK,IAAI,EAAI,EAAG,EAAI,EAAO,OAAQ,IAC/B,EAAM,GAAK,EAAO,GAEtB,OAAO,IAAI,EACP,EACA,EACA,EACA,EAAQ,WAAa,CAAC,IAAK,GAAG,EAC9B,EAAQ,MAAQ,EAChB,EAAQ,KAAO,EACf,EAAQ,cAAgB,EAC5B,CACJ,CAGA,IAAI,QAA4B,CAC5B,OAAO,KAAK,OAChB,CAGA,IAAI,OAA0C,CAC1C,OAAO,KAAK,MAChB,CAGA,IAAI,YAAqB,CACrB,OAAO,KAAK,QAAQ,MACxB,CAGA,MAAM,KACF,EACA,EAAoC,CAAC,EACL,CAChC,OAAO,KAAK,QAAQ,EAAO,CAAO,CACtC,CAGA,MAAM,QACF,EACA,EAAoC,CAAC,EACL,CAChC,IAAM,EAAO,OAAO,GAAU,SAAW,EAAQ,KAC3C,EAAW,MAAM,EAAA,UAAU,CAAK,EAChC,EAAS,KAAK,YAAY,CAAQ,EAClC,EAAU,MAAM,KAAK,SAAS,IAAI,EAAG,KAAK,SAAS,WAAY,CAAO,CAAC,EACvE,EAAkB,KAAK,SAAS,YAAY,GAClD,GAAI,IAAoB,IAAA,GACpB,MAAU,MAAM,kCAAkC,EAEtD,IAAM,EAAM,EAAQ,GACpB,GAAI,IAAQ,IAAA,GACR,MAAU,MACN,2BAA2B,EAAgB,4BAC/C,EAEJ,IAAM,EAAY,KAAK,aAAa,EAAI,IAAoB,EAEtD,CAAE,UAAS,UAAW,EAAA,KAAK,EAAW,EAAQ,MAAQ,IAAI,EAC1D,EAAoC,CAAC,EAC3C,IAAK,IAAI,EAAI,EAAG,EAAI,EAAQ,OAAQ,IAAK,CACrC,IAAM,EAAK,EAAQ,GACb,EAAY,KAAK,QAAQ,IAAO,SAAS,IAC/C,EAAc,KAAK,CACf,QAAS,EACT,YACA,YAAa,EAAO,GACpB,IAAK,EACL,KAAM,EACN,KAAM,EAAO,EACjB,CAAC,CACL,CACA,GAAI,EAAc,SAAW,EACzB,MAAU,MAAM,sDAAsD,EAG1E,IAAM,EAAM,EAAc,GACpB,EAA+B,CACjC,QAAS,EAAI,QACb,UAAW,EAAI,UACf,WAAY,EAAI,YAChB,IAAK,EAAI,QACT,KAAM,EAAI,UACV,KAAM,EAAI,YACV,MAAO,EACP,eACJ,EAEM,EAAkC,CAAC,EAAS,OAAQ,EAAS,KAAK,EACxE,MAAO,CACH,IAAI,EAAA,sBACA,IAAI,EAAA,MAAM,CAAS,EACnB,EACA,KAAK,OACL,EACA,EACA,CACJ,CACJ,CACJ,CAEA,YAAoB,EAA6B,CAC7C,GAAM,CAAC,EAAI,GAAM,KAAK,WAChB,EAAU,EAAA,OAAO,EAAO,EAAI,CAAE,EAGpC,OAAO,EAAA,gBADK,EAAA,MADO,EAAA,UAAU,EAAS,KAAK,MAAO,KAAK,IACrC,EAAY,EAAQ,MAAO,EAAQ,OAAQ,CACtC,EAAK,CAAC,EAAG,EAAG,EAAQ,OAAQ,EAAQ,KAAK,CAAC,CACrE,CAEA,aAAqB,EAAiC,CAClD,OAAO,KAAK,cAAgB,EAAA,QAAQ,CAAG,EAAI,IAAI,aAAa,CAAG,CACnE,CACJ"}

package/dist/vision/tasks/classifier.js

@@ -0,0 +1,90 @@
+import { ClassificationResults as e, Probs as t } from "../results.js";
+import { resolveLabels as n } from "../labels.js";
+import { OrtSession as r } from "../core/session.js";
+import { loadImage as i } from "../io/image.js";
+import { normalize as a, resize as o, toCHW as s, toFloat32Tensor as c } from "../preprocess/image.js";
+import { softmax as l, topK as u } from "../postprocess/classification.js";
+import { VisionTask as d } from "./base.js";
+//#region src/vision/tasks/classifier.ts
+var f = [
+	.485,
+	.456,
+	.406
+], p = [
+	.229,
+	.224,
+	.225
+], m = class m extends d {
+	_labels;
+	_names;
+	_inputSize;
+	_mean;
+	_std;
+	_applySoftmax;
+	constructor(e, t, n, r, i, a, o) {
+		super(e), this._labels = t, this._names = n, this._inputSize = r, this._mean = i, this._std = a, this._applySoftmax = o;
+	}
+	static async create(e, t) {
+		let i = await r.create(e, t), a = n(t.labels, { numClasses: t.numClasses }), o = {};
+		for (let e = 0; e < a.length; e++) o[e] = a[e];
+		return new m(i, a, o, t.inputSize ?? [224, 224], t.mean ?? f, t.std ?? p, t.applySoftmax ?? !0);
+	}
+	get labels() {
+		return this._labels;
+	}
+	get names() {
+		return this._names;
+	}
+	get numClasses() {
+		return this._labels.length;
+	}
+	async call(e, t = {}) {
+		return this.predict(e, t);
+	}
+	async predict(n, r = {}) {
+		let a = typeof n == "string" ? n : null, o = await i(n), s = this._preprocess(o), c = await this._session.run({ [this._session.inputName]: s }), l = this._session.outputNames[0];
+		if (l === void 0) throw Error("Classifier model has no outputs.");
+		let d = c[l];
+		if (d === void 0) throw Error(`Classifier model output ${l} missing from run() result.`);
+		let f = this._postprocess(d.data), { indices: p, values: m } = u(f, r.topK ?? null), h = [];
+		for (let e = 0; e < p.length; e++) {
+			let t = p[e], n = this._labels[t] ?? `class_${t}`;
+			h.push({
+				classId: t,
+				className: n,
+				probability: m[e],
+				cls: t,
+				name: n,
+				conf: m[e]
+			});
+		}
+		if (h.length === 0) throw Error("Classifier produced no probabilities (empty output).");
+		let g = h[0], _ = {
+			classId: g.classId,
+			className: g.className,
+			confidence: g.probability,
+			cls: g.classId,
+			name: g.className,
+			conf: g.probability,
+			image: o,
+			probabilities: h
+		}, v = [o.height, o.width];
+		return [new e(new t(f), _, this._names, o, v, a)];
+	}
+	_preprocess(e) {
+		let [t, n] = this._inputSize, r = o(e, t, n);
+		return c(s(a(r, this._mean, this._std), r.width, r.height, 3), [
+			1,
+			3,
+			r.height,
+			r.width
+		]);
+	}
+	_postprocess(e) {
+		return this._applySoftmax ? l(e) : new Float32Array(e);
+	}
+};
+//#endregion
+export { m as Classifier };
+
+//# sourceMappingURL=classifier.js.map

package/dist/vision/tasks/classifier.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"classifier.js","names":[],"sources":["../../../src/vision/tasks/classifier.ts"],"sourcesContent":["/**\n * Image classification task using ONNX Runtime Web.\n */\n\nimport type * as ort from \"onnxruntime-web\";\n\nimport { type ModelSource, type OrtSessionOptions, OrtSession } from \"../core/session\";\nimport { type ImageInput, loadImage } from \"../io/image\";\nimport { type LabelSpec, resolveLabels } from \"../labels\";\nimport { softmax, topK } from \"../postprocess/classification\";\nimport { normalize, resize, toCHW, toFloat32Tensor } from \"../preprocess/image\";\nimport { ClassificationResults, Probs } from \"../results\";\nimport { VisionTask } from \"./base\";\nimport { type ClassProbability, type ClassificationResult, type RGBImage } from \"../types\";\n\nconst IMAGENET_MEAN: readonly [number, number, number] = [0.485, 0.456, 0.406];\nconst IMAGENET_STD: readonly [number, number, number] = [0.229, 0.224, 0.225];\n\nexport interface ClassifierOptions extends OrtSessionOptions {\n    /** Class label spec — see {@link resolveLabels}. */\n    readonly labels: LabelSpec;\n    /**\n     * Number of classes the model can predict. Required when `labels` is `null`\n     * or when you want to validate that the supplied labels match the model.\n     */\n    readonly numClasses?: number;\n    /** Model input `[width, height]` in pixels. Defaults to `[224, 224]`. */\n    readonly inputSize?: readonly [number, number];\n    /** Per-channel RGB mean used for normalization. Defaults to ImageNet. */\n    readonly mean?: readonly [number, number, number];\n    /** Per-channel RGB standard deviation. Defaults to ImageNet. */\n    readonly std?: readonly [number, number, number];\n    /**\n     * If `true` (default), apply softmax to the raw model output. Set to\n     * `false` for models whose final layer already produces a probability\n     * distribution.\n     */\n    readonly applySoftmax?: boolean;\n}\n\nexport interface ClassifierPredictOptions {\n    /**\n     * If set, the per-class probability list in `results[0].result.probabilities`\n     * is truncated to the top-K entries. The bulk `probs` view always exposes\n     * the full vector.\n     */\n    readonly topK?: number;\n}\n\n/**\n * Image classifier wrapping an ONNX model with ImageNet-style preprocessing.\n *\n * `predict()` returns `Promise<ClassificationResults[]>` (length 1 for a\n * single image), mirroring Ultralytics' API. The envelope exposes a `probs`\n * collection (`top1`, `top1conf`, `top5`, `top5conf`, `data`) plus the\n * legacy per-class probability list with names already resolved.\n *\n * Defaults: 224×224 RGB input, `float32` normalized with ImageNet mean/std,\n * NCHW layout, batch size 1, softmax applied to the raw output.\n *\n * @example\n * ```typescript\n * const clf = await Classifier.create(\"/models/resnet50.onnx\", {\n *   labels: [\"tench\", \"goldfish\", ...]  // 1000 ImageNet labels\n * });\n * const r = (await clf.predict(\"/images/dog.jpg\"))[0];\n * console.log(r.cls, r.conf, r.name);\n * console.log(r.probs.top5, r.probs.top5conf);\n * ```\n */\nexport class Classifier extends VisionTask {\n    private constructor(\n        session: OrtSession,\n        private readonly _labels: readonly string[],\n        private readonly _names: Readonly<Record<number, string>>,\n        private readonly _inputSize: readonly [number, number],\n        private readonly _mean: readonly [number, number, number],\n        private readonly _std: readonly [number, number, number],\n        private readonly _applySoftmax: boolean,\n    ) {\n        super(session);\n    }\n\n    /** Load the model and resolve labels. */\n    static async create(model: ModelSource, options: ClassifierOptions): Promise<Classifier> {\n        const session = await OrtSession.create(model, options);\n        const labels = resolveLabels(options.labels, { numClasses: options.numClasses });\n        const names: Record<number, string> = {};\n        for (let i = 0; i < labels.length; i++) {\n            names[i] = labels[i] as string;\n        }\n        return new Classifier(\n            session,\n            labels,\n            names,\n            options.inputSize ?? [224, 224],\n            options.mean ?? IMAGENET_MEAN,\n            options.std ?? IMAGENET_STD,\n            options.applySoftmax ?? true,\n        );\n    }\n\n    /** Class labels indexed by class id. */\n    get labels(): readonly string[] {\n        return this._labels;\n    }\n\n    /** Class id → class name dict (matches Ultralytics' `model.names`). */\n    get names(): Readonly<Record<number, string>> {\n        return this._names;\n    }\n\n    /** Number of classes the model can predict. */\n    get numClasses(): number {\n        return this._labels.length;\n    }\n\n    /** Alias for {@link predict} (parity with PyTorch `nn.Module.__call__`). */\n    async call(\n        image: ImageInput,\n        options: ClassifierPredictOptions = {},\n    ): Promise<ClassificationResults[]> {\n        return this.predict(image, options);\n    }\n\n    /** Run classification on a single image. */\n    async predict(\n        image: ImageInput,\n        options: ClassifierPredictOptions = {},\n    ): Promise<ClassificationResults[]> {\n        const path = typeof image === \"string\" ? image : null;\n        const original = await loadImage(image);\n        const tensor = this._preprocess(original);\n        const outputs = await this._session.run({ [this._session.inputName]: tensor });\n        const firstOutputName = this._session.outputNames[0];\n        if (firstOutputName === undefined) {\n            throw new Error(\"Classifier model has no outputs.\");\n        }\n        const raw = outputs[firstOutputName];\n        if (raw === undefined) {\n            throw new Error(\n                `Classifier model output ${firstOutputName} missing from run() result.`,\n            );\n        }\n        const fullProbs = this._postprocess(raw.data as Float32Array);\n\n        const { indices, values } = topK(fullProbs, options.topK ?? null);\n        const probabilities: ClassProbability[] = [];\n        for (let i = 0; i < indices.length; i++) {\n            const id = indices[i] as number;\n            const className = this._labels[id] ?? `class_${id}`;\n            probabilities.push({\n                classId: id,\n                className,\n                probability: values[i] as number,\n                cls: id,\n                name: className,\n                conf: values[i] as number,\n            });\n        }\n        if (probabilities.length === 0) {\n            throw new Error(\"Classifier produced no probabilities (empty output).\");\n        }\n\n        const top = probabilities[0] as ClassProbability;\n        const result: ClassificationResult = {\n            classId: top.classId,\n            className: top.className,\n            confidence: top.probability,\n            cls: top.classId,\n            name: top.className,\n            conf: top.probability,\n            image: original,\n            probabilities,\n        };\n\n        const orig: readonly [number, number] = [original.height, original.width];\n        return [\n            new ClassificationResults(\n                new Probs(fullProbs),\n                result,\n                this._names,\n                original,\n                orig,\n                path,\n            ),\n        ];\n    }\n\n    private _preprocess(image: RGBImage): ort.Tensor {\n        const [tw, th] = this._inputSize;\n        const resized = resize(image, tw, th);\n        const normalized = normalize(resized, this._mean, this._std);\n        const chw = toCHW(normalized, resized.width, resized.height, 3);\n        return toFloat32Tensor(chw, [1, 3, resized.height, resized.width]);\n    }\n\n    private _postprocess(raw: Float32Array): Float32Array {\n        return this._applySoftmax ? softmax(raw) : new Float32Array(raw);\n    }\n}\n"],"mappings":";;;;;;;;AAeA,IAAM,IAAmD;CAAC;CAAO;CAAO;AAAK,GACvE,IAAkD;CAAC;CAAO;CAAO;AAAK,GAsD/D,IAAb,MAAa,UAAmB,EAAW;CAGlB;CACA;CACA;CACA;CACA;CACA;CAPrB,YACI,GACA,GACA,GACA,GACA,GACA,GACA,GACF;EADmB,AAEjB,MAAM,CAAO,GAPI,KAAA,UAAA,GACA,KAAA,SAAA,GACA,KAAA,aAAA,GACA,KAAA,QAAA,GACA,KAAA,OAAA,GACA,KAAA,gBAAA;CAGrB;CAGA,aAAa,OAAO,GAAoB,GAAiD;EACrF,IAAM,IAAU,MAAM,EAAW,OAAO,GAAO,CAAO,GAChD,IAAS,EAAc,EAAQ,QAAQ,EAAE,YAAY,EAAQ,WAAW,CAAC,GACzE,IAAgC,CAAC;EACvC,KAAK,IAAI,IAAI,GAAG,IAAI,EAAO,QAAQ,KAC/B,EAAM,KAAK,EAAO;EAEtB,OAAO,IAAI,EACP,GACA,GACA,GACA,EAAQ,aAAa,CAAC,KAAK,GAAG,GAC9B,EAAQ,QAAQ,GAChB,EAAQ,OAAO,GACf,EAAQ,gBAAgB,EAC5B;CACJ;CAGA,IAAI,SAA4B;EAC5B,OAAO,KAAK;CAChB;CAGA,IAAI,QAA0C;EAC1C,OAAO,KAAK;CAChB;CAGA,IAAI,aAAqB;EACrB,OAAO,KAAK,QAAQ;CACxB;CAGA,MAAM,KACF,GACA,IAAoC,CAAC,GACL;EAChC,OAAO,KAAK,QAAQ,GAAO,CAAO;CACtC;CAGA,MAAM,QACF,GACA,IAAoC,CAAC,GACL;EAChC,IAAM,IAAO,OAAO,KAAU,WAAW,IAAQ,MAC3C,IAAW,MAAM,EAAU,CAAK,GAChC,IAAS,KAAK,YAAY,CAAQ,GAClC,IAAU,MAAM,KAAK,SAAS,IAAI,GAAG,KAAK,SAAS,YAAY,EAAO,CAAC,GACvE,IAAkB,KAAK,SAAS,YAAY;EAClD,IAAI,MAAoB,KAAA,GACpB,MAAU,MAAM,kCAAkC;EAEtD,IAAM,IAAM,EAAQ;EACpB,IAAI,MAAQ,KAAA,GACR,MAAU,MACN,2BAA2B,EAAgB,4BAC/C;EAEJ,IAAM,IAAY,KAAK,aAAa,EAAI,IAAoB,GAEtD,EAAE,YAAS,cAAW,EAAK,GAAW,EAAQ,QAAQ,IAAI,GAC1D,IAAoC,CAAC;EAC3C,KAAK,IAAI,IAAI,GAAG,IAAI,EAAQ,QAAQ,KAAK;GACrC,IAAM,IAAK,EAAQ,IACb,IAAY,KAAK,QAAQ,MAAO,SAAS;GAC/C,EAAc,KAAK;IACf,SAAS;IACT;IACA,aAAa,EAAO;IACpB,KAAK;IACL,MAAM;IACN,MAAM,EAAO;GACjB,CAAC;EACL;EACA,IAAI,EAAc,WAAW,GACzB,MAAU,MAAM,sDAAsD;EAG1E,IAAM,IAAM,EAAc,IACpB,IAA+B;GACjC,SAAS,EAAI;GACb,WAAW,EAAI;GACf,YAAY,EAAI;GAChB,KAAK,EAAI;GACT,MAAM,EAAI;GACV,MAAM,EAAI;GACV,OAAO;GACP;EACJ,GAEM,IAAkC,CAAC,EAAS,QAAQ,EAAS,KAAK;EACxE,OAAO,CACH,IAAI,EACA,IAAI,EAAM,CAAS,GACnB,GACA,KAAK,QACL,GACA,GACA,CACJ,CACJ;CACJ;CAEA,YAAoB,GAA6B;EAC7C,IAAM,CAAC,GAAI,KAAM,KAAK,YAChB,IAAU,EAAO,GAAO,GAAI,CAAE;EAGpC,OAAO,EADK,EADO,EAAU,GAAS,KAAK,OAAO,KAAK,IACrC,GAAY,EAAQ,OAAO,EAAQ,QAAQ,CACtC,GAAK;GAAC;GAAG;GAAG,EAAQ;GAAQ,EAAQ;EAAK,CAAC;CACrE;CAEA,aAAqB,GAAiC;EAClD,OAAO,KAAK,gBAAgB,EAAQ,CAAG,IAAI,IAAI,aAAa,CAAG;CACnE;AACJ"}

package/dist/vision/tasks/detector.cjs

@@ -0,0 +1,2 @@
+const e=require("../types.cjs"),t=require("../results.cjs"),n=require("../labels.cjs"),r=require("../core/session.cjs"),i=require("../io/image.cjs"),a=require("../preprocess/image.cjs"),o=require("../postprocess/detection.cjs"),s=require("./base.cjs");var c=class c extends s.VisionTask{_head;_labels;_names;_inputSize;_confThreshold;_iouThreshold;_maxDetections;constructor(e,t,n,r,i,a,o,s){super(e),this._head=t,this._labels=n,this._names=r,this._inputSize=i,this._confThreshold=a,this._iouThreshold=o,this._maxDetections=s}static async create(e,t={}){let i=t.head??`yolo`;if(i!==`yolo`)throw Error(`Unsupported detector head '${i}'. Supported: 'yolo'.`);let a=await r.OrtSession.create(e,t),o=n.resolveLabels(t.labels??`coco`,{numClasses:t.numClasses}),s={};for(let e=0;e<o.length;e++)s[e]=o[e];return new c(a,i,o,s,t.inputSize??[640,640],t.confThreshold??.25,t.iouThreshold??.45,t.maxDetections??300)}get head(){return this._head}get labels(){return this._labels}get names(){return this._names}get numClasses(){return this._labels.length}async call(e,t={}){return this.predict(e,t)}async predict(e,n={}){let r=typeof e==`string`?e:null,a=await i.loadImage(e),{tensor:s,scale:c,padLeft:l,padTop:u}=this._preprocess(a),d=await this._session.run({[this._session.inputName]:s}),f=this._session.outputNames[0];if(f===void 0)throw Error(`Detector model has no outputs.`);let p=d[f];if(p===void 0)throw Error(`Detector model output ${f} missing from run() result.`);let m=o.decodeYolo(p.data,p.dims,{originalWidth:a.width,originalHeight:a.height,padLeft:l,padTop:u,scale:c,confThreshold:n.confThreshold??this._confThreshold,iouThreshold:n.iouThreshold??this._iouThreshold,maxDetections:this._maxDetections}),h=(n.classes===void 0?m:(()=>{let e=new Set(n.classes);return m.filter(t=>e.has(t.classId))})()).map(e=>this._buildResult(a,e.bbox,e.classId,e.confidence)),g=[a.height,a.width];return[new t.DetectionResults(this._buildBoxes(h,g),h,this._names,a,g,r)]}_preprocess(e){let[t,n]=this._inputSize,r=a.letterbox(e,t,n);return{tensor:a.toFloat32Tensor(a.toCHW(a.toFloat32(r.image),r.image.width,r.image.height,3),[1,3,r.image.height,r.image.width]),scale:r.scale,padLeft:r.padLeft,padTop:r.padTop}}_buildResult(t,n,r,i){let[a,o,s,c]=n.asIntXyxy(),l=Math.max(0,a),u=Math.max(0,o),d=Math.min(t.width,s),f=Math.min(t.height,c),p;if(d>l&&f>u){let n=d-l,r=f-u,i=new Uint8Array(n*r*3);for(let e=0;e<r;e++){let r=((u+e)*t.width+l)*3;i.set(t.data.subarray(r,r+n*3),e*n*3)}p=new e.RGBImage(i,n,r)}else p=new e.RGBImage(new Uint8Array,0,0);let m=this._names[r]??`class_${r}`;return{classId:r,className:m,confidence:i,bbox:n,cls:r,name:m,conf:i,box:n,croppedImage:p}}_buildBoxes(e,n){let r=e.length,i=new Float32Array(r*4),a=new Int32Array(r),o=new Float32Array(r);for(let t=0;t<r;t++){let n=e[t];i[t*4]=n.bbox.x1,i[t*4+1]=n.bbox.y1,i[t*4+2]=n.bbox.x2,i[t*4+3]=n.bbox.y2,a[t]=n.classId,o[t]=n.confidence}return new t.Boxes(i,a,o,n)}};exports.Detector=c;
+//# sourceMappingURL=detector.cjs.map

package/dist/vision/tasks/detector.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"detector.cjs","names":[],"sources":["../../../src/vision/tasks/detector.ts"],"sourcesContent":["/**\n * Object detection task using anchor-free YOLO ONNX models (v8/v9/v10/v11/v12).\n */\n\nimport type * as ort from \"onnxruntime-web\";\n\nimport { type ModelSource, type OrtSessionOptions, OrtSession } from \"../core/session\";\nimport { type ImageInput, loadImage } from \"../io/image\";\nimport { type LabelSpec, resolveLabels } from \"../labels\";\nimport { decodeYolo } from \"../postprocess/detection\";\nimport { letterbox, toCHW, toFloat32, toFloat32Tensor } from \"../preprocess/image\";\nimport { Boxes, DetectionResults } from \"../results\";\nimport { VisionTask } from \"./base\";\nimport { type BoundingBox, type DetectionResult, RGBImage } from \"../types\";\n\n/**\n * Decoder family for the model's detection head.\n *\n * - `\"yolo\"`: anchor-free YOLO head with output shape `[1, 4 + nc, N]` —\n *   covers YOLOv8, v9, v10, v11, v12, v26 detect exports.\n *\n * The SDK does **not** auto-detect the head from the model — the caller is\n * responsible for picking a head that matches their export. Future families\n * (v5/v6/v7 with `[1, N, 5+nc]`) will be added as new literal members.\n */\nexport type DetectorHead = \"yolo\";\n\nexport interface DetectorOptions extends OrtSessionOptions {\n    /**\n     * Decoder family for the detection head. Default `\"yolo\"` covers\n     * YOLOv8/v9/v10/v11/v12/v26.\n     */\n    readonly head?: DetectorHead;\n    /** Class label spec — see {@link resolveLabels}. Defaults to the COCO 80-class preset. */\n    readonly labels?: LabelSpec;\n    /** Number of classes — used to validate the supplied labels. */\n    readonly numClasses?: number;\n    /** Model input `[width, height]` for letterboxing. Defaults to `[640, 640]`. */\n    readonly inputSize?: readonly [number, number];\n    /** Default minimum class score to keep a candidate. */\n    readonly confThreshold?: number;\n    /** Default IoU threshold for non-maximum suppression. */\n    readonly iouThreshold?: number;\n    /** Maximum number of detections per image. */\n    readonly maxDetections?: number;\n}\n\nexport interface DetectorPredictOptions {\n    /** Override the default confidence threshold. */\n    readonly confThreshold?: number;\n    /** Override the default IoU threshold. */\n    readonly iouThreshold?: number;\n    /**\n     * If set, keep only detections whose `classId` is in this list.\n     * Mirrors Ultralytics' `model.predict(img, classes=[0, 16])`.\n     */\n    readonly classes?: readonly number[];\n}\n\n/**\n * Object detector for anchor-free YOLO ONNX models (v8/v9/v10/v11/v12).\n *\n * `predict()` returns `Promise<DetectionResults[]>` (length 1 for a single\n * image), mirroring Ultralytics' `YOLO(\"img.jpg\")`. Iterate the envelope for\n * per-instance dataclasses, or use the bulk `boxes` view (`.xyxy`, `.xywh`,\n * `.xyxyn`, `.xywhn`, `.cls`, `.conf`).\n *\n * @example\n * ```typescript\n * const det = await Detector.create(\"/models/yolov8n.onnx\");\n * const results = await det.predict(\"/images/street.jpg\");\n * const r = results[0];\n * console.log(r.boxes.xyxy, r.boxes.cls, r.boxes.conf, r.names);\n * for (const d of r) {\n *   console.log(d.cls, d.conf, d.box.xyxy);\n * }\n * ```\n */\nexport class Detector extends VisionTask {\n    private constructor(\n        session: OrtSession,\n        private readonly _head: DetectorHead,\n        private readonly _labels: readonly string[],\n        private readonly _names: Readonly<Record<number, string>>,\n        private readonly _inputSize: readonly [number, number],\n        private readonly _confThreshold: number,\n        private readonly _iouThreshold: number,\n        private readonly _maxDetections: number,\n    ) {\n        super(session);\n    }\n\n    /** Load the model and resolve labels. */\n    static async create(model: ModelSource, options: DetectorOptions = {}): Promise<Detector> {\n        const head: DetectorHead = options.head ?? \"yolo\";\n        if (head !== \"yolo\") {\n            throw new Error(`Unsupported detector head '${head}'. Supported: 'yolo'.`);\n        }\n        const session = await OrtSession.create(model, options);\n        const labels = resolveLabels(options.labels ?? \"coco\", {\n            numClasses: options.numClasses,\n        });\n        const names: Record<number, string> = {};\n        for (let i = 0; i < labels.length; i++) {\n            names[i] = labels[i] as string;\n        }\n        return new Detector(\n            session,\n            head,\n            labels,\n            names,\n            options.inputSize ?? [640, 640],\n            options.confThreshold ?? 0.25,\n            options.iouThreshold ?? 0.45,\n            options.maxDetections ?? 300,\n        );\n    }\n\n    /** The decoder family used to interpret the model's output. */\n    get head(): DetectorHead {\n        return this._head;\n    }\n\n    /** Class labels indexed by class id. */\n    get labels(): readonly string[] {\n        return this._labels;\n    }\n\n    /** Class id → class name dict (matches Ultralytics' `model.names`). */\n    get names(): Readonly<Record<number, string>> {\n        return this._names;\n    }\n\n    /** Number of classes the model predicts. */\n    get numClasses(): number {\n        return this._labels.length;\n    }\n\n    /**\n     * Alias for {@link predict} — call the detector like a torch `nn.Module`.\n     *\n     * Use as `det.call(img)` since JavaScript class instances are not callable;\n     * for direct invocation, prefer `det.predict(img)`. The full\n     * {@link DetectorPredictOptions} (including `classes`) is supported.\n     */\n    async call(\n        image: ImageInput,\n        options: DetectorPredictOptions = {},\n    ): Promise<DetectionResults[]> {\n        return this.predict(image, options);\n    }\n\n    /** Run detection on a single image. */\n    async predict(\n        image: ImageInput,\n        options: DetectorPredictOptions = {},\n    ): Promise<DetectionResults[]> {\n        const path = typeof image === \"string\" ? image : null;\n        const original = await loadImage(image);\n        const { tensor, scale, padLeft, padTop } = this._preprocess(original);\n        const outputs = await this._session.run({ [this._session.inputName]: tensor });\n\n        const firstOutputName = this._session.outputNames[0];\n        if (firstOutputName === undefined) {\n            throw new Error(\"Detector model has no outputs.\");\n        }\n        const raw = outputs[firstOutputName];\n        if (raw === undefined) {\n            throw new Error(`Detector model output ${firstOutputName} missing from run() result.`);\n        }\n\n        const decodedAll = decodeYolo(raw.data as Float32Array, raw.dims, {\n            originalWidth: original.width,\n            originalHeight: original.height,\n            padLeft,\n            padTop,\n            scale,\n            confThreshold: options.confThreshold ?? this._confThreshold,\n            iouThreshold: options.iouThreshold ?? this._iouThreshold,\n            maxDetections: this._maxDetections,\n        });\n\n        const decoded =\n            options.classes !== undefined\n                ? (() => {\n                      const allowed = new Set(options.classes);\n                      return decodedAll.filter((d) => allowed.has(d.classId));\n                  })()\n                : decodedAll;\n\n        const detections = decoded.map((d) =>\n            this._buildResult(original, d.bbox, d.classId, d.confidence),\n        );\n\n        const orig: readonly [number, number] = [original.height, original.width];\n        return [\n            new DetectionResults(\n                this._buildBoxes(detections, orig),\n                detections,\n                this._names,\n                original,\n                orig,\n                path,\n            ),\n        ];\n    }\n\n    private _preprocess(image: RGBImage): {\n        tensor: ort.Tensor;\n        scale: number;\n        padLeft: number;\n        padTop: number;\n    } {\n        const [tw, th] = this._inputSize;\n        const lb = letterbox(image, tw, th);\n        const f32 = toFloat32(lb.image);\n        const chw = toCHW(f32, lb.image.width, lb.image.height, 3);\n        return {\n            tensor: toFloat32Tensor(chw, [1, 3, lb.image.height, lb.image.width]),\n            scale: lb.scale,\n            padLeft: lb.padLeft,\n            padTop: lb.padTop,\n        };\n    }\n\n    private _buildResult(\n        original: RGBImage,\n        bbox: BoundingBox,\n        classId: number,\n        confidence: number,\n    ): DetectionResult {\n        const [x1, y1, x2, y2] = bbox.asIntXyxy();\n        const cx1 = Math.max(0, x1);\n        const cy1 = Math.max(0, y1);\n        const cx2 = Math.min(original.width, x2);\n        const cy2 = Math.min(original.height, y2);\n\n        let cropped: RGBImage;\n        if (cx2 > cx1 && cy2 > cy1) {\n            const cw = cx2 - cx1;\n            const ch = cy2 - cy1;\n            const out = new Uint8Array(cw * ch * 3);\n            for (let row = 0; row < ch; row++) {\n                const srcOffset = ((cy1 + row) * original.width + cx1) * 3;\n                out.set(original.data.subarray(srcOffset, srcOffset + cw * 3), row * cw * 3);\n            }\n            cropped = new RGBImage(out, cw, ch);\n        } else {\n            cropped = new RGBImage(new Uint8Array(0), 0, 0);\n        }\n\n        const className = this._names[classId] ?? `class_${classId}`;\n\n        return {\n            classId,\n            className,\n            confidence,\n            bbox,\n            cls: classId,\n            name: className,\n            conf: confidence,\n            box: bbox,\n            croppedImage: cropped,\n        };\n    }\n\n    private _buildBoxes(\n        detections: readonly DetectionResult[],\n        origShape: readonly [number, number],\n    ): Boxes {\n        const n = detections.length;\n        const xyxy = new Float32Array(n * 4);\n        const cls = new Int32Array(n);\n        const conf = new Float32Array(n);\n        for (let i = 0; i < n; i++) {\n            const d = detections[i] as DetectionResult;\n            xyxy[i * 4] = d.bbox.x1;\n            xyxy[i * 4 + 1] = d.bbox.y1;\n            xyxy[i * 4 + 2] = d.bbox.x2;\n            xyxy[i * 4 + 3] = d.bbox.y2;\n            cls[i] = d.classId;\n            conf[i] = d.confidence;\n        }\n        return new Boxes(xyxy, cls, conf, origShape);\n    }\n}\n"],"mappings":"4PA8EA,IAAa,EAAb,MAAa,UAAiB,EAAA,UAAW,CAGhB,MACA,QACA,OACA,WACA,eACA,cACA,eARrB,YACI,EACA,EACA,EACA,EACA,EACA,EACA,EACA,EACF,CACE,MAAM,CAAO,EARI,KAAA,MAAA,EACA,KAAA,QAAA,EACA,KAAA,OAAA,EACA,KAAA,WAAA,EACA,KAAA,eAAA,EACA,KAAA,cAAA,EACA,KAAA,eAAA,CAGrB,CAGA,aAAa,OAAO,EAAoB,EAA2B,CAAC,EAAsB,CACtF,IAAM,EAAqB,EAAQ,MAAQ,OAC3C,GAAI,IAAS,OACT,MAAU,MAAM,8BAA8B,EAAK,sBAAsB,EAE7E,IAAM,EAAU,MAAM,EAAA,WAAW,OAAO,EAAO,CAAO,EAChD,EAAS,EAAA,cAAc,EAAQ,QAAU,OAAQ,CACnD,WAAY,EAAQ,UACxB,CAAC,EACK,EAAgC,CAAC,EACvC,IAAK,IAAI,EAAI,EAAG,EAAI,EAAO,OAAQ,IAC/B,EAAM,GAAK,EAAO,GAEtB,OAAO,IAAI,EACP,EACA,EACA,EACA,EACA,EAAQ,WAAa,CAAC,IAAK,GAAG,EAC9B,EAAQ,eAAiB,IACzB,EAAQ,cAAgB,IACxB,EAAQ,eAAiB,GAC7B,CACJ,CAGA,IAAI,MAAqB,CACrB,OAAO,KAAK,KAChB,CAGA,IAAI,QAA4B,CAC5B,OAAO,KAAK,OAChB,CAGA,IAAI,OAA0C,CAC1C,OAAO,KAAK,MAChB,CAGA,IAAI,YAAqB,CACrB,OAAO,KAAK,QAAQ,MACxB,CASA,MAAM,KACF,EACA,EAAkC,CAAC,EACR,CAC3B,OAAO,KAAK,QAAQ,EAAO,CAAO,CACtC,CAGA,MAAM,QACF,EACA,EAAkC,CAAC,EACR,CAC3B,IAAM,EAAO,OAAO,GAAU,SAAW,EAAQ,KAC3C,EAAW,MAAM,EAAA,UAAU,CAAK,EAChC,CAAE,SAAQ,QAAO,UAAS,UAAW,KAAK,YAAY,CAAQ,EAC9D,EAAU,MAAM,KAAK,SAAS,IAAI,EAAG,KAAK,SAAS,WAAY,CAAO,CAAC,EAEvE,EAAkB,KAAK,SAAS,YAAY,GAClD,GAAI,IAAoB,IAAA,GACpB,MAAU,MAAM,gCAAgC,EAEpD,IAAM,EAAM,EAAQ,GACpB,GAAI,IAAQ,IAAA,GACR,MAAU,MAAM,yBAAyB,EAAgB,4BAA4B,EAGzF,IAAM,EAAa,EAAA,WAAW,EAAI,KAAsB,EAAI,KAAM,CAC9D,cAAe,EAAS,MACxB,eAAgB,EAAS,OACzB,UACA,SACA,QACA,cAAe,EAAQ,eAAiB,KAAK,eAC7C,aAAc,EAAQ,cAAgB,KAAK,cAC3C,cAAe,KAAK,cACxB,CAAC,EAUK,GAPF,EAAQ,UAAY,IAAA,GAKd,OAJO,CACH,IAAM,EAAU,IAAI,IAAI,EAAQ,OAAO,EACvC,OAAO,EAAW,OAAQ,GAAM,EAAQ,IAAI,EAAE,OAAO,CAAC,CAC1D,EAAA,CAAG,EACH,CAEiB,IAAK,GAC5B,KAAK,aAAa,EAAU,EAAE,KAAM,EAAE,QAAS,EAAE,UAAU,CAC/D,EAEM,EAAkC,CAAC,EAAS,OAAQ,EAAS,KAAK,EACxE,MAAO,CACH,IAAI,EAAA,iBACA,KAAK,YAAY,EAAY,CAAI,EACjC,EACA,KAAK,OACL,EACA,EACA,CACJ,CACJ,CACJ,CAEA,YAAoB,EAKlB,CACE,GAAM,CAAC,EAAI,GAAM,KAAK,WAChB,EAAK,EAAA,UAAU,EAAO,EAAI,CAAE,EAGlC,MAAO,CACH,OAAQ,EAAA,gBAFA,EAAA,MADA,EAAA,UAAU,EAAG,KACP,EAAK,EAAG,MAAM,MAAO,EAAG,MAAM,OAAQ,CAE5B,EAAK,CAAC,EAAG,EAAG,EAAG,MAAM,OAAQ,EAAG,MAAM,KAAK,CAAC,EACpE,MAAO,EAAG,MACV,QAAS,EAAG,QACZ,OAAQ,EAAG,MACf,CACJ,CAEA,aACI,EACA,EACA,EACA,EACe,CACf,GAAM,CAAC,EAAI,EAAI,EAAI,GAAM,EAAK,UAAU,EAClC,EAAM,KAAK,IAAI,EAAG,CAAE,EACpB,EAAM,KAAK,IAAI,EAAG,CAAE,EACpB,EAAM,KAAK,IAAI,EAAS,MAAO,CAAE,EACjC,EAAM,KAAK,IAAI,EAAS,OAAQ,CAAE,EAEpC,EACJ,GAAI,EAAM,GAAO,EAAM,EAAK,CACxB,IAAM,EAAK,EAAM,EACX,EAAK,EAAM,EACX,EAAM,IAAI,WAAW,EAAK,EAAK,CAAC,EACtC,IAAK,IAAI,EAAM,EAAG,EAAM,EAAI,IAAO,CAC/B,IAAM,IAAc,EAAM,GAAO,EAAS,MAAQ,GAAO,EACzD,EAAI,IAAI,EAAS,KAAK,SAAS,EAAW,EAAY,EAAK,CAAC,EAAG,EAAM,EAAK,CAAC,CAC/E,CACA,EAAU,IAAI,EAAA,SAAS,EAAK,EAAI,CAAE,CACtC,KACI,GAAU,IAAI,EAAA,SAAS,IAAI,WAAe,EAAG,CAAC,EAGlD,IAAM,EAAY,KAAK,OAAO,IAAY,SAAS,IAEnD,MAAO,CACH,UACA,YACA,aACA,OACA,IAAK,EACL,KAAM,EACN,KAAM,EACN,IAAK,EACL,aAAc,CAClB,CACJ,CAEA,YACI,EACA,EACK,CACL,IAAM,EAAI,EAAW,OACf,EAAO,IAAI,aAAa,EAAI,CAAC,EAC7B,EAAM,IAAI,WAAW,CAAC,EACtB,EAAO,IAAI,aAAa,CAAC,EAC/B,IAAK,IAAI,EAAI,EAAG,EAAI,EAAG,IAAK,CACxB,IAAM,EAAI,EAAW,GACrB,EAAK,EAAI,GAAK,EAAE,KAAK,GACrB,EAAK,EAAI,EAAI,GAAK,EAAE,KAAK,GACzB,EAAK,EAAI,EAAI,GAAK,EAAE,KAAK,GACzB,EAAK,EAAI,EAAI,GAAK,EAAE,KAAK,GACzB,EAAI,GAAK,EAAE,QACX,EAAK,GAAK,EAAE,UAChB,CACA,OAAO,IAAI,EAAA,MAAM,EAAM,EAAK,EAAM,CAAS,CAC/C,CACJ"}

package/dist/vision/tasks/detector.js

@@ -0,0 +1,112 @@
+import { RGBImage as e } from "../types.js";
+import { Boxes as t, DetectionResults as n } from "../results.js";
+import { resolveLabels as r } from "../labels.js";
+import { OrtSession as i } from "../core/session.js";
+import { loadImage as a } from "../io/image.js";
+import { letterbox as o, toCHW as s, toFloat32 as c, toFloat32Tensor as l } from "../preprocess/image.js";
+import { decodeYolo as u } from "../postprocess/detection.js";
+import { VisionTask as d } from "./base.js";
+//#region src/vision/tasks/detector.ts
+var f = class f extends d {
+	_head;
+	_labels;
+	_names;
+	_inputSize;
+	_confThreshold;
+	_iouThreshold;
+	_maxDetections;
+	constructor(e, t, n, r, i, a, o, s) {
+		super(e), this._head = t, this._labels = n, this._names = r, this._inputSize = i, this._confThreshold = a, this._iouThreshold = o, this._maxDetections = s;
+	}
+	static async create(e, t = {}) {
+		let n = t.head ?? "yolo";
+		if (n !== "yolo") throw Error(`Unsupported detector head '${n}'. Supported: 'yolo'.`);
+		let a = await i.create(e, t), o = r(t.labels ?? "coco", { numClasses: t.numClasses }), s = {};
+		for (let e = 0; e < o.length; e++) s[e] = o[e];
+		return new f(a, n, o, s, t.inputSize ?? [640, 640], t.confThreshold ?? .25, t.iouThreshold ?? .45, t.maxDetections ?? 300);
+	}
+	get head() {
+		return this._head;
+	}
+	get labels() {
+		return this._labels;
+	}
+	get names() {
+		return this._names;
+	}
+	get numClasses() {
+		return this._labels.length;
+	}
+	async call(e, t = {}) {
+		return this.predict(e, t);
+	}
+	async predict(e, t = {}) {
+		let r = typeof e == "string" ? e : null, i = await a(e), { tensor: o, scale: s, padLeft: c, padTop: l } = this._preprocess(i), d = await this._session.run({ [this._session.inputName]: o }), f = this._session.outputNames[0];
+		if (f === void 0) throw Error("Detector model has no outputs.");
+		let p = d[f];
+		if (p === void 0) throw Error(`Detector model output ${f} missing from run() result.`);
+		let m = u(p.data, p.dims, {
+			originalWidth: i.width,
+			originalHeight: i.height,
+			padLeft: c,
+			padTop: l,
+			scale: s,
+			confThreshold: t.confThreshold ?? this._confThreshold,
+			iouThreshold: t.iouThreshold ?? this._iouThreshold,
+			maxDetections: this._maxDetections
+		}), h = (t.classes === void 0 ? m : (() => {
+			let e = new Set(t.classes);
+			return m.filter((t) => e.has(t.classId));
+		})()).map((e) => this._buildResult(i, e.bbox, e.classId, e.confidence)), g = [i.height, i.width];
+		return [new n(this._buildBoxes(h, g), h, this._names, i, g, r)];
+	}
+	_preprocess(e) {
+		let [t, n] = this._inputSize, r = o(e, t, n);
+		return {
+			tensor: l(s(c(r.image), r.image.width, r.image.height, 3), [
+				1,
+				3,
+				r.image.height,
+				r.image.width
+			]),
+			scale: r.scale,
+			padLeft: r.padLeft,
+			padTop: r.padTop
+		};
+	}
+	_buildResult(t, n, r, i) {
+		let [a, o, s, c] = n.asIntXyxy(), l = Math.max(0, a), u = Math.max(0, o), d = Math.min(t.width, s), f = Math.min(t.height, c), p;
+		if (d > l && f > u) {
+			let n = d - l, r = f - u, i = new Uint8Array(n * r * 3);
+			for (let e = 0; e < r; e++) {
+				let r = ((u + e) * t.width + l) * 3;
+				i.set(t.data.subarray(r, r + n * 3), e * n * 3);
+			}
+			p = new e(i, n, r);
+		} else p = new e(/* @__PURE__ */ new Uint8Array(), 0, 0);
+		let m = this._names[r] ?? `class_${r}`;
+		return {
+			classId: r,
+			className: m,
+			confidence: i,
+			bbox: n,
+			cls: r,
+			name: m,
+			conf: i,
+			box: n,
+			croppedImage: p
+		};
+	}
+	_buildBoxes(e, n) {
+		let r = e.length, i = new Float32Array(r * 4), a = new Int32Array(r), o = new Float32Array(r);
+		for (let t = 0; t < r; t++) {
+			let n = e[t];
+			i[t * 4] = n.bbox.x1, i[t * 4 + 1] = n.bbox.y1, i[t * 4 + 2] = n.bbox.x2, i[t * 4 + 3] = n.bbox.y2, a[t] = n.classId, o[t] = n.confidence;
+		}
+		return new t(i, a, o, n);
+	}
+};
+//#endregion
+export { f as Detector };
+
+//# sourceMappingURL=detector.js.map

package/dist/vision/tasks/detector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"detector.js","names":[],"sources":["../../../src/vision/tasks/detector.ts"],"sourcesContent":["/**\n * Object detection task using anchor-free YOLO ONNX models (v8/v9/v10/v11/v12).\n */\n\nimport type * as ort from \"onnxruntime-web\";\n\nimport { type ModelSource, type OrtSessionOptions, OrtSession } from \"../core/session\";\nimport { type ImageInput, loadImage } from \"../io/image\";\nimport { type LabelSpec, resolveLabels } from \"../labels\";\nimport { decodeYolo } from \"../postprocess/detection\";\nimport { letterbox, toCHW, toFloat32, toFloat32Tensor } from \"../preprocess/image\";\nimport { Boxes, DetectionResults } from \"../results\";\nimport { VisionTask } from \"./base\";\nimport { type BoundingBox, type DetectionResult, RGBImage } from \"../types\";\n\n/**\n * Decoder family for the model's detection head.\n *\n * - `\"yolo\"`: anchor-free YOLO head with output shape `[1, 4 + nc, N]` —\n *   covers YOLOv8, v9, v10, v11, v12, v26 detect exports.\n *\n * The SDK does **not** auto-detect the head from the model — the caller is\n * responsible for picking a head that matches their export. Future families\n * (v5/v6/v7 with `[1, N, 5+nc]`) will be added as new literal members.\n */\nexport type DetectorHead = \"yolo\";\n\nexport interface DetectorOptions extends OrtSessionOptions {\n    /**\n     * Decoder family for the detection head. Default `\"yolo\"` covers\n     * YOLOv8/v9/v10/v11/v12/v26.\n     */\n    readonly head?: DetectorHead;\n    /** Class label spec — see {@link resolveLabels}. Defaults to the COCO 80-class preset. */\n    readonly labels?: LabelSpec;\n    /** Number of classes — used to validate the supplied labels. */\n    readonly numClasses?: number;\n    /** Model input `[width, height]` for letterboxing. Defaults to `[640, 640]`. */\n    readonly inputSize?: readonly [number, number];\n    /** Default minimum class score to keep a candidate. */\n    readonly confThreshold?: number;\n    /** Default IoU threshold for non-maximum suppression. */\n    readonly iouThreshold?: number;\n    /** Maximum number of detections per image. */\n    readonly maxDetections?: number;\n}\n\nexport interface DetectorPredictOptions {\n    /** Override the default confidence threshold. */\n    readonly confThreshold?: number;\n    /** Override the default IoU threshold. */\n    readonly iouThreshold?: number;\n    /**\n     * If set, keep only detections whose `classId` is in this list.\n     * Mirrors Ultralytics' `model.predict(img, classes=[0, 16])`.\n     */\n    readonly classes?: readonly number[];\n}\n\n/**\n * Object detector for anchor-free YOLO ONNX models (v8/v9/v10/v11/v12).\n *\n * `predict()` returns `Promise<DetectionResults[]>` (length 1 for a single\n * image), mirroring Ultralytics' `YOLO(\"img.jpg\")`. Iterate the envelope for\n * per-instance dataclasses, or use the bulk `boxes` view (`.xyxy`, `.xywh`,\n * `.xyxyn`, `.xywhn`, `.cls`, `.conf`).\n *\n * @example\n * ```typescript\n * const det = await Detector.create(\"/models/yolov8n.onnx\");\n * const results = await det.predict(\"/images/street.jpg\");\n * const r = results[0];\n * console.log(r.boxes.xyxy, r.boxes.cls, r.boxes.conf, r.names);\n * for (const d of r) {\n *   console.log(d.cls, d.conf, d.box.xyxy);\n * }\n * ```\n */\nexport class Detector extends VisionTask {\n    private constructor(\n        session: OrtSession,\n        private readonly _head: DetectorHead,\n        private readonly _labels: readonly string[],\n        private readonly _names: Readonly<Record<number, string>>,\n        private readonly _inputSize: readonly [number, number],\n        private readonly _confThreshold: number,\n        private readonly _iouThreshold: number,\n        private readonly _maxDetections: number,\n    ) {\n        super(session);\n    }\n\n    /** Load the model and resolve labels. */\n    static async create(model: ModelSource, options: DetectorOptions = {}): Promise<Detector> {\n        const head: DetectorHead = options.head ?? \"yolo\";\n        if (head !== \"yolo\") {\n            throw new Error(`Unsupported detector head '${head}'. Supported: 'yolo'.`);\n        }\n        const session = await OrtSession.create(model, options);\n        const labels = resolveLabels(options.labels ?? \"coco\", {\n            numClasses: options.numClasses,\n        });\n        const names: Record<number, string> = {};\n        for (let i = 0; i < labels.length; i++) {\n            names[i] = labels[i] as string;\n        }\n        return new Detector(\n            session,\n            head,\n            labels,\n            names,\n            options.inputSize ?? [640, 640],\n            options.confThreshold ?? 0.25,\n            options.iouThreshold ?? 0.45,\n            options.maxDetections ?? 300,\n        );\n    }\n\n    /** The decoder family used to interpret the model's output. */\n    get head(): DetectorHead {\n        return this._head;\n    }\n\n    /** Class labels indexed by class id. */\n    get labels(): readonly string[] {\n        return this._labels;\n    }\n\n    /** Class id → class name dict (matches Ultralytics' `model.names`). */\n    get names(): Readonly<Record<number, string>> {\n        return this._names;\n    }\n\n    /** Number of classes the model predicts. */\n    get numClasses(): number {\n        return this._labels.length;\n    }\n\n    /**\n     * Alias for {@link predict} — call the detector like a torch `nn.Module`.\n     *\n     * Use as `det.call(img)` since JavaScript class instances are not callable;\n     * for direct invocation, prefer `det.predict(img)`. The full\n     * {@link DetectorPredictOptions} (including `classes`) is supported.\n     */\n    async call(\n        image: ImageInput,\n        options: DetectorPredictOptions = {},\n    ): Promise<DetectionResults[]> {\n        return this.predict(image, options);\n    }\n\n    /** Run detection on a single image. */\n    async predict(\n        image: ImageInput,\n        options: DetectorPredictOptions = {},\n    ): Promise<DetectionResults[]> {\n        const path = typeof image === \"string\" ? image : null;\n        const original = await loadImage(image);\n        const { tensor, scale, padLeft, padTop } = this._preprocess(original);\n        const outputs = await this._session.run({ [this._session.inputName]: tensor });\n\n        const firstOutputName = this._session.outputNames[0];\n        if (firstOutputName === undefined) {\n            throw new Error(\"Detector model has no outputs.\");\n        }\n        const raw = outputs[firstOutputName];\n        if (raw === undefined) {\n            throw new Error(`Detector model output ${firstOutputName} missing from run() result.`);\n        }\n\n        const decodedAll = decodeYolo(raw.data as Float32Array, raw.dims, {\n            originalWidth: original.width,\n            originalHeight: original.height,\n            padLeft,\n            padTop,\n            scale,\n            confThreshold: options.confThreshold ?? this._confThreshold,\n            iouThreshold: options.iouThreshold ?? this._iouThreshold,\n            maxDetections: this._maxDetections,\n        });\n\n        const decoded =\n            options.classes !== undefined\n                ? (() => {\n                      const allowed = new Set(options.classes);\n                      return decodedAll.filter((d) => allowed.has(d.classId));\n                  })()\n                : decodedAll;\n\n        const detections = decoded.map((d) =>\n            this._buildResult(original, d.bbox, d.classId, d.confidence),\n        );\n\n        const orig: readonly [number, number] = [original.height, original.width];\n        return [\n            new DetectionResults(\n                this._buildBoxes(detections, orig),\n                detections,\n                this._names,\n                original,\n                orig,\n                path,\n            ),\n        ];\n    }\n\n    private _preprocess(image: RGBImage): {\n        tensor: ort.Tensor;\n        scale: number;\n        padLeft: number;\n        padTop: number;\n    } {\n        const [tw, th] = this._inputSize;\n        const lb = letterbox(image, tw, th);\n        const f32 = toFloat32(lb.image);\n        const chw = toCHW(f32, lb.image.width, lb.image.height, 3);\n        return {\n            tensor: toFloat32Tensor(chw, [1, 3, lb.image.height, lb.image.width]),\n            scale: lb.scale,\n            padLeft: lb.padLeft,\n            padTop: lb.padTop,\n        };\n    }\n\n    private _buildResult(\n        original: RGBImage,\n        bbox: BoundingBox,\n        classId: number,\n        confidence: number,\n    ): DetectionResult {\n        const [x1, y1, x2, y2] = bbox.asIntXyxy();\n        const cx1 = Math.max(0, x1);\n        const cy1 = Math.max(0, y1);\n        const cx2 = Math.min(original.width, x2);\n        const cy2 = Math.min(original.height, y2);\n\n        let cropped: RGBImage;\n        if (cx2 > cx1 && cy2 > cy1) {\n            const cw = cx2 - cx1;\n            const ch = cy2 - cy1;\n            const out = new Uint8Array(cw * ch * 3);\n            for (let row = 0; row < ch; row++) {\n                const srcOffset = ((cy1 + row) * original.width + cx1) * 3;\n                out.set(original.data.subarray(srcOffset, srcOffset + cw * 3), row * cw * 3);\n            }\n            cropped = new RGBImage(out, cw, ch);\n        } else {\n            cropped = new RGBImage(new Uint8Array(0), 0, 0);\n        }\n\n        const className = this._names[classId] ?? `class_${classId}`;\n\n        return {\n            classId,\n            className,\n            confidence,\n            bbox,\n            cls: classId,\n            name: className,\n            conf: confidence,\n            box: bbox,\n            croppedImage: cropped,\n        };\n    }\n\n    private _buildBoxes(\n        detections: readonly DetectionResult[],\n        origShape: readonly [number, number],\n    ): Boxes {\n        const n = detections.length;\n        const xyxy = new Float32Array(n * 4);\n        const cls = new Int32Array(n);\n        const conf = new Float32Array(n);\n        for (let i = 0; i < n; i++) {\n            const d = detections[i] as DetectionResult;\n            xyxy[i * 4] = d.bbox.x1;\n            xyxy[i * 4 + 1] = d.bbox.y1;\n            xyxy[i * 4 + 2] = d.bbox.x2;\n            xyxy[i * 4 + 3] = d.bbox.y2;\n            cls[i] = d.classId;\n            conf[i] = d.confidence;\n        }\n        return new Boxes(xyxy, cls, conf, origShape);\n    }\n}\n"],"mappings":";;;;;;;;;AA8EA,IAAa,IAAb,MAAa,UAAiB,EAAW;CAGhB;CACA;CACA;CACA;CACA;CACA;CACA;CARrB,YACI,GACA,GACA,GACA,GACA,GACA,GACA,GACA,GACF;EADmB,AAEjB,MAAM,CAAO,GARI,KAAA,QAAA,GACA,KAAA,UAAA,GACA,KAAA,SAAA,GACA,KAAA,aAAA,GACA,KAAA,iBAAA,GACA,KAAA,gBAAA,GACA,KAAA,iBAAA;CAGrB;CAGA,aAAa,OAAO,GAAoB,IAA2B,CAAC,GAAsB;EACtF,IAAM,IAAqB,EAAQ,QAAQ;EAC3C,IAAI,MAAS,QACT,MAAU,MAAM,8BAA8B,EAAK,sBAAsB;EAE7E,IAAM,IAAU,MAAM,EAAW,OAAO,GAAO,CAAO,GAChD,IAAS,EAAc,EAAQ,UAAU,QAAQ,EACnD,YAAY,EAAQ,WACxB,CAAC,GACK,IAAgC,CAAC;EACvC,KAAK,IAAI,IAAI,GAAG,IAAI,EAAO,QAAQ,KAC/B,EAAM,KAAK,EAAO;EAEtB,OAAO,IAAI,EACP,GACA,GACA,GACA,GACA,EAAQ,aAAa,CAAC,KAAK,GAAG,GAC9B,EAAQ,iBAAiB,KACzB,EAAQ,gBAAgB,KACxB,EAAQ,iBAAiB,GAC7B;CACJ;CAGA,IAAI,OAAqB;EACrB,OAAO,KAAK;CAChB;CAGA,IAAI,SAA4B;EAC5B,OAAO,KAAK;CAChB;CAGA,IAAI,QAA0C;EAC1C,OAAO,KAAK;CAChB;CAGA,IAAI,aAAqB;EACrB,OAAO,KAAK,QAAQ;CACxB;CASA,MAAM,KACF,GACA,IAAkC,CAAC,GACR;EAC3B,OAAO,KAAK,QAAQ,GAAO,CAAO;CACtC;CAGA,MAAM,QACF,GACA,IAAkC,CAAC,GACR;EAC3B,IAAM,IAAO,OAAO,KAAU,WAAW,IAAQ,MAC3C,IAAW,MAAM,EAAU,CAAK,GAChC,EAAE,WAAQ,UAAO,YAAS,cAAW,KAAK,YAAY,CAAQ,GAC9D,IAAU,MAAM,KAAK,SAAS,IAAI,GAAG,KAAK,SAAS,YAAY,EAAO,CAAC,GAEvE,IAAkB,KAAK,SAAS,YAAY;EAClD,IAAI,MAAoB,KAAA,GACpB,MAAU,MAAM,gCAAgC;EAEpD,IAAM,IAAM,EAAQ;EACpB,IAAI,MAAQ,KAAA,GACR,MAAU,MAAM,yBAAyB,EAAgB,4BAA4B;EAGzF,IAAM,IAAa,EAAW,EAAI,MAAsB,EAAI,MAAM;GAC9D,eAAe,EAAS;GACxB,gBAAgB,EAAS;GACzB;GACA;GACA;GACA,eAAe,EAAQ,iBAAiB,KAAK;GAC7C,cAAc,EAAQ,gBAAgB,KAAK;GAC3C,eAAe,KAAK;EACxB,CAAC,GAUK,KAPF,EAAQ,YAAY,KAAA,IAKd,WAJO;GACH,IAAM,IAAU,IAAI,IAAI,EAAQ,OAAO;GACvC,OAAO,EAAW,QAAQ,MAAM,EAAQ,IAAI,EAAE,OAAO,CAAC;EAC1D,EAAA,CAAG,EACH,CAEiB,KAAK,MAC5B,KAAK,aAAa,GAAU,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,CAC/D,GAEM,IAAkC,CAAC,EAAS,QAAQ,EAAS,KAAK;EACxE,OAAO,CACH,IAAI,EACA,KAAK,YAAY,GAAY,CAAI,GACjC,GACA,KAAK,QACL,GACA,GACA,CACJ,CACJ;CACJ;CAEA,YAAoB,GAKlB;EACE,IAAM,CAAC,GAAI,KAAM,KAAK,YAChB,IAAK,EAAU,GAAO,GAAI,CAAE;EAGlC,OAAO;GACH,QAAQ,EAFA,EADA,EAAU,EAAG,KACP,GAAK,EAAG,MAAM,OAAO,EAAG,MAAM,QAAQ,CAE5B,GAAK;IAAC;IAAG;IAAG,EAAG,MAAM;IAAQ,EAAG,MAAM;GAAK,CAAC;GACpE,OAAO,EAAG;GACV,SAAS,EAAG;GACZ,QAAQ,EAAG;EACf;CACJ;CAEA,aACI,GACA,GACA,GACA,GACe;EACf,IAAM,CAAC,GAAI,GAAI,GAAI,KAAM,EAAK,UAAU,GAClC,IAAM,KAAK,IAAI,GAAG,CAAE,GACpB,IAAM,KAAK,IAAI,GAAG,CAAE,GACpB,IAAM,KAAK,IAAI,EAAS,OAAO,CAAE,GACjC,IAAM,KAAK,IAAI,EAAS,QAAQ,CAAE,GAEpC;EACJ,IAAI,IAAM,KAAO,IAAM,GAAK;GACxB,IAAM,IAAK,IAAM,GACX,IAAK,IAAM,GACX,IAAM,IAAI,WAAW,IAAK,IAAK,CAAC;GACtC,KAAK,IAAI,IAAM,GAAG,IAAM,GAAI,KAAO;IAC/B,IAAM,MAAc,IAAM,KAAO,EAAS,QAAQ,KAAO;IACzD,EAAI,IAAI,EAAS,KAAK,SAAS,GAAW,IAAY,IAAK,CAAC,GAAG,IAAM,IAAK,CAAC;GAC/E;GACA,IAAU,IAAI,EAAS,GAAK,GAAI,CAAE;EACtC,OACI,IAAU,IAAI,kBAAS,IAAI,WAAY,GAAG,GAAG,CAAC;EAGlD,IAAM,IAAY,KAAK,OAAO,MAAY,SAAS;EAEnD,OAAO;GACH;GACA;GACA;GACA;GACA,KAAK;GACL,MAAM;GACN,MAAM;GACN,KAAK;GACL,cAAc;EAClB;CACJ;CAEA,YACI,GACA,GACK;EACL,IAAM,IAAI,EAAW,QACf,IAAO,IAAI,aAAa,IAAI,CAAC,GAC7B,IAAM,IAAI,WAAW,CAAC,GACtB,IAAO,IAAI,aAAa,CAAC;EAC/B,KAAK,IAAI,IAAI,GAAG,IAAI,GAAG,KAAK;GACxB,IAAM,IAAI,EAAW;GAMrB,AALA,EAAK,IAAI,KAAK,EAAE,KAAK,IACrB,EAAK,IAAI,IAAI,KAAK,EAAE,KAAK,IACzB,EAAK,IAAI,IAAI,KAAK,EAAE,KAAK,IACzB,EAAK,IAAI,IAAI,KAAK,EAAE,KAAK,IACzB,EAAI,KAAK,EAAE,SACX,EAAK,KAAK,EAAE;EAChB;EACA,OAAO,IAAI,EAAM,GAAM,GAAK,GAAM,CAAS;CAC/C;AACJ"}

package/dist/vision/tasks/segmenter.cjs

@@ -0,0 +1,2 @@
+const e=require("../types.cjs"),t=require("../results.cjs"),n=require("../labels.cjs"),r=require("../core/session.cjs"),i=require("../io/image.cjs"),a=require("../preprocess/image.cjs"),o=require("../postprocess/segmentation.cjs"),s=require("./base.cjs");var c=class c extends s.VisionTask{_head;_labels;_names;_inputSize;_confThreshold;_iouThreshold;_maxDetections;_maskThreshold;constructor(e,t,n,r,i,a,o,s,c){super(e),this._head=t,this._labels=n,this._names=r,this._inputSize=i,this._confThreshold=a,this._iouThreshold=o,this._maxDetections=s,this._maskThreshold=c}static async create(e,t={}){let i=t.head??`yolo-seg`;if(i!==`yolo-seg`)throw Error(`Unsupported segmenter head '${i}'. Supported: 'yolo-seg'.`);let a=await r.OrtSession.create(e,t),o=n.resolveLabels(t.labels??`coco`,{numClasses:t.numClasses}),s={};for(let e=0;e<o.length;e++)s[e]=o[e];return new c(a,i,o,s,t.inputSize??[640,640],t.confThreshold??.25,t.iouThreshold??.45,t.maxDetections??300,t.maskThreshold??.5)}get head(){return this._head}get labels(){return this._labels}get names(){return this._names}get numClasses(){return this._labels.length}async call(e,t={}){return this.predict(e,t)}async predict(e,n={}){let r=typeof e==`string`?e:null,a=await i.loadImage(e),{tensor:s,scale:c,padLeft:l,padTop:u}=this._preprocess(a),d=await this._session.run({[this._session.inputName]:s}),{perAnchor:f,prototypes:p}=this._splitOutputs(d),m=o.decodeYoloSeg(f.data,f.dims,p.data,p.dims,{numClasses:this._labels.length,inputWidth:this._inputSize[0],inputHeight:this._inputSize[1],originalWidth:a.width,originalHeight:a.height,padLeft:l,padTop:u,scale:c,confThreshold:n.confThreshold??this._confThreshold,iouThreshold:n.iouThreshold??this._iouThreshold,maxDetections:this._maxDetections,maskThreshold:this._maskThreshold}),h=(n.classes===void 0?m:(()=>{let e=new Set(n.classes);return m.filter(t=>e.has(t.classId))})()).map(e=>this._buildResult(a,e.bbox,e.classId,e.confidence,e.mask)),g=[a.height,a.width];return[new t.SegmentationResults(this._buildBoxes(h,g),this._buildMasks(h,g),h,this._names,a,g,r)]}_preprocess(e){let[t,n]=this._inputSize,r=a.letterbox(e,t,n);return{tensor:a.toFloat32Tensor(a.toCHW(a.toFloat32(r.image),r.image.width,r.image.height,3),[1,3,r.image.height,r.image.width]),scale:r.scale,padLeft:r.padLeft,padTop:r.padTop}}_splitOutputs(e){let t,n;for(let r of this._session.outputNames){let i=e[r];i!==void 0&&(i.dims.length===3&&t===void 0?t=i:i.dims.length===4&&n===void 0&&(n=i))}if(t===void 0||n===void 0){let t=this._session.outputNames.map(t=>`${t}: ${JSON.stringify(e[t]?.dims??[])}`);throw Error(`Segmenter expected one 3-D and one 4-D output, got [${t.join(`, `)}].`)}return{perAnchor:t,prototypes:n}}_buildResult(t,n,r,i,a){let[o,s,c,l]=n.asIntXyxy(),u=Math.max(0,o),d=Math.max(0,s),f=Math.min(t.width,c),p=Math.min(t.height,l),m,h=a;if(f>u&&p>d&&a.data.length>0){let n=f-u,r=p-d,i=Math.min(a.width,n),o=Math.min(a.height,r),s=new Uint8Array(i*o*3);for(let e=0;e<o;e++){let n=((d+e)*t.width+u)*3,r=e*i*3,o=e*a.width;for(let e=0;e<i;e++)if(a.data[o+e]!==0){let i=n+e*3,a=r+e*3;s[a]=t.data[i],s[a+1]=t.data[i+1],s[a+2]=t.data[i+2]}}if(m=new e.RGBImage(s,i,o),i!==a.width||o!==a.height){let t=new Uint8Array(i*o);for(let e=0;e<o;e++)t.set(a.data.subarray(e*a.width,e*a.width+i),e*i);h=new e.Mask(t,i,o)}}else h=new e.Mask(new Uint8Array,0,0),m=new e.RGBImage(new Uint8Array,0,0);let g=this._names[r]??`class_${r}`;return{classId:r,className:g,confidence:i,bbox:n,cls:r,name:g,conf:i,box:n,mask:h,segmentedImage:m}}_buildBoxes(e,n){let r=e.length,i=new Float32Array(r*4),a=new Int32Array(r),o=new Float32Array(r);for(let t=0;t<r;t++){let n=e[t];i[t*4]=n.bbox.x1,i[t*4+1]=n.bbox.y1,i[t*4+2]=n.bbox.x2,i[t*4+3]=n.bbox.y2,a[t]=n.classId,o[t]=n.confidence}return new t.Boxes(i,a,o,n)}_buildMasks(e,n){let r=new Float32Array(e.length*4);for(let t=0;t<e.length;t++){let n=e[t];r[t*4]=n.bbox.x1,r[t*4+1]=n.bbox.y1,r[t*4+2]=n.bbox.x2,r[t*4+3]=n.bbox.y2}return new t.Masks(e.map(e=>e.mask),r,n)}};exports.Segmenter=c;
+//# sourceMappingURL=segmenter.cjs.map

package/dist/vision/tasks/segmenter.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"segmenter.cjs","names":[],"sources":["../../../src/vision/tasks/segmenter.ts"],"sourcesContent":["/**\n * Instance-segmentation task using YOLO seg ONNX models (v8-seg / v11-seg / ...).\n */\n\nimport type * as ort from \"onnxruntime-web\";\n\nimport { type ModelSource, type OrtSessionOptions, OrtSession } from \"../core/session\";\nimport { type ImageInput, loadImage } from \"../io/image\";\nimport { type LabelSpec, resolveLabels } from \"../labels\";\nimport { decodeYoloSeg } from \"../postprocess/segmentation\";\nimport { letterbox, toCHW, toFloat32, toFloat32Tensor } from \"../preprocess/image\";\nimport { Boxes, Masks, SegmentationResults } from \"../results\";\nimport { VisionTask } from \"./base\";\nimport { type BoundingBox, type SegmentationResult, Mask, RGBImage } from \"../types\";\n\n/**\n * Decoder family for the segmentation head.\n *\n * - `\"yolo-seg\"`: YOLO instance-segmentation head with two outputs —\n *   `[1, 4 + nc + nm, N]` per-anchor predictions plus `[1, nm, mh, mw]`\n *   prototype masks. Covers YOLOv8-seg, v11-seg, v26-seg.\n *\n * The SDK does **not** auto-detect this — the caller is responsible for\n * picking a head that matches their export.\n */\nexport type SegmenterHead = \"yolo-seg\";\n\nexport interface SegmenterOptions extends OrtSessionOptions {\n    /**\n     * Decoder family for the segmentation head. Default `\"yolo-seg\"` covers\n     * YOLOv8-seg/v11-seg/v26-seg.\n     */\n    readonly head?: SegmenterHead;\n    /** Class label spec — see {@link resolveLabels}. Defaults to the COCO 80-class preset. */\n    readonly labels?: LabelSpec;\n    /** Number of classes — used to validate the supplied labels. */\n    readonly numClasses?: number;\n    /** Model input `[width, height]` for letterboxing. Defaults to `[640, 640]`. */\n    readonly inputSize?: readonly [number, number];\n    /** Default minimum class score to keep a candidate. */\n    readonly confThreshold?: number;\n    /** Default IoU threshold for non-maximum suppression. */\n    readonly iouThreshold?: number;\n    /** Maximum number of instances per image. */\n    readonly maxDetections?: number;\n    /** Probability cutoff applied to soft masks. Defaults to `0.5`. */\n    readonly maskThreshold?: number;\n}\n\nexport interface SegmenterPredictOptions {\n    readonly confThreshold?: number;\n    readonly iouThreshold?: number;\n    /**\n     * If set, keep only instances whose `classId` is in this list.\n     * Mirrors Ultralytics' `model.predict(img, classes=[0, 16])`.\n     */\n    readonly classes?: readonly number[];\n}\n\n/**\n * Instance segmenter for YOLO seg ONNX models (v8-seg / v11-seg / ...).\n *\n * The model is expected to expose two outputs:\n *\n * 1. `output0`: `(1, 4 + numClasses + numMaskCoefs, numAnchors)` — per-anchor\n *    predictions (boxes, class scores, mask coefficients).\n * 2. `output1`: `(1, numMaskCoefs, maskH, maskW)` — prototype masks.\n *\n * `predict()` returns `Promise<SegmentationResults[]>` (length 1 for a\n * single image), mirroring Ultralytics' API. The envelope exposes:\n *\n * - `boxes`: bulk numpy view (`xyxy`, `xywh`, `xyxyn`, `xywhn`, `cls`, `conf`).\n * - `masks`: per-instance binary masks cropped to each box.\n * - per-instance {@link SegmentationResult} via iteration.\n *\n * @example\n * ```typescript\n * const seg = await Segmenter.create(\"/models/yolov8n-seg.onnx\");\n * const r = (await seg.predict(\"/images/street.jpg\"))[0];\n * for (const inst of r) {\n *   console.log(inst.cls, inst.conf, inst.box.xyxy);\n * }\n * ```\n */\nexport class Segmenter extends VisionTask {\n    private constructor(\n        session: OrtSession,\n        private readonly _head: SegmenterHead,\n        private readonly _labels: readonly string[],\n        private readonly _names: Readonly<Record<number, string>>,\n        private readonly _inputSize: readonly [number, number],\n        private readonly _confThreshold: number,\n        private readonly _iouThreshold: number,\n        private readonly _maxDetections: number,\n        private readonly _maskThreshold: number,\n    ) {\n        super(session);\n    }\n\n    /** Load the model and resolve labels. */\n    static async create(model: ModelSource, options: SegmenterOptions = {}): Promise<Segmenter> {\n        const head: SegmenterHead = options.head ?? \"yolo-seg\";\n        if (head !== \"yolo-seg\") {\n            throw new Error(`Unsupported segmenter head '${head}'. Supported: 'yolo-seg'.`);\n        }\n        const session = await OrtSession.create(model, options);\n        const labels = resolveLabels(options.labels ?? \"coco\", {\n            numClasses: options.numClasses,\n        });\n        const names: Record<number, string> = {};\n        for (let i = 0; i < labels.length; i++) {\n            names[i] = labels[i] as string;\n        }\n        return new Segmenter(\n            session,\n            head,\n            labels,\n            names,\n            options.inputSize ?? [640, 640],\n            options.confThreshold ?? 0.25,\n            options.iouThreshold ?? 0.45,\n            options.maxDetections ?? 300,\n            options.maskThreshold ?? 0.5,\n        );\n    }\n\n    /** The decoder family used to interpret the model's output. */\n    get head(): SegmenterHead {\n        return this._head;\n    }\n\n    /** Class labels indexed by class id. */\n    get labels(): readonly string[] {\n        return this._labels;\n    }\n\n    /** Class id → class name dict (matches Ultralytics' `model.names`). */\n    get names(): Readonly<Record<number, string>> {\n        return this._names;\n    }\n\n    /** Number of classes the model predicts. */\n    get numClasses(): number {\n        return this._labels.length;\n    }\n\n    /** Alias for {@link predict} (parity with PyTorch `nn.Module.__call__`). */\n    async call(\n        image: ImageInput,\n        options: SegmenterPredictOptions = {},\n    ): Promise<SegmentationResults[]> {\n        return this.predict(image, options);\n    }\n\n    /** Run instance segmentation on a single image. */\n    async predict(\n        image: ImageInput,\n        options: SegmenterPredictOptions = {},\n    ): Promise<SegmentationResults[]> {\n        const path = typeof image === \"string\" ? image : null;\n        const original = await loadImage(image);\n        const { tensor, scale, padLeft, padTop } = this._preprocess(original);\n        const outputs = await this._session.run({ [this._session.inputName]: tensor });\n\n        const { perAnchor, prototypes } = this._splitOutputs(outputs);\n\n        const decodedAll = decodeYoloSeg(\n            perAnchor.data as Float32Array,\n            perAnchor.dims,\n            prototypes.data as Float32Array,\n            prototypes.dims,\n            {\n                numClasses: this._labels.length,\n                inputWidth: this._inputSize[0],\n                inputHeight: this._inputSize[1],\n                originalWidth: original.width,\n                originalHeight: original.height,\n                padLeft,\n                padTop,\n                scale,\n                confThreshold: options.confThreshold ?? this._confThreshold,\n                iouThreshold: options.iouThreshold ?? this._iouThreshold,\n                maxDetections: this._maxDetections,\n                maskThreshold: this._maskThreshold,\n            },\n        );\n\n        const decoded =\n            options.classes !== undefined\n                ? (() => {\n                      const allowed = new Set(options.classes);\n                      return decodedAll.filter((d) => allowed.has(d.classId));\n                  })()\n                : decodedAll;\n\n        const detections = decoded.map((d) =>\n            this._buildResult(original, d.bbox, d.classId, d.confidence, d.mask),\n        );\n\n        const orig: readonly [number, number] = [original.height, original.width];\n        return [\n            new SegmentationResults(\n                this._buildBoxes(detections, orig),\n                this._buildMasks(detections, orig),\n                detections,\n                this._names,\n                original,\n                orig,\n                path,\n            ),\n        ];\n    }\n\n    private _preprocess(image: RGBImage): {\n        tensor: ort.Tensor;\n        scale: number;\n        padLeft: number;\n        padTop: number;\n    } {\n        const [tw, th] = this._inputSize;\n        const lb = letterbox(image, tw, th);\n        const f32 = toFloat32(lb.image);\n        const chw = toCHW(f32, lb.image.width, lb.image.height, 3);\n        return {\n            tensor: toFloat32Tensor(chw, [1, 3, lb.image.height, lb.image.width]),\n            scale: lb.scale,\n            padLeft: lb.padLeft,\n            padTop: lb.padTop,\n        };\n    }\n\n    private _splitOutputs(outputs: Record<string, ort.Tensor>): {\n        perAnchor: ort.Tensor;\n        prototypes: ort.Tensor;\n    } {\n        let perAnchor: ort.Tensor | undefined;\n        let prototypes: ort.Tensor | undefined;\n        for (const name of this._session.outputNames) {\n            const t = outputs[name];\n            if (t === undefined) continue;\n            if (t.dims.length === 3 && perAnchor === undefined) {\n                perAnchor = t;\n            } else if (t.dims.length === 4 && prototypes === undefined) {\n                prototypes = t;\n            }\n        }\n        if (perAnchor === undefined || prototypes === undefined) {\n            const shapes = this._session.outputNames.map(\n                (n) => `${n}: ${JSON.stringify(outputs[n]?.dims ?? [])}`,\n            );\n            throw new Error(\n                `Segmenter expected one 3-D and one 4-D output, got [${shapes.join(\", \")}].`,\n            );\n        }\n        return { perAnchor, prototypes };\n    }\n\n    private _buildResult(\n        original: RGBImage,\n        bbox: BoundingBox,\n        classId: number,\n        confidence: number,\n        mask: Mask,\n    ): SegmentationResult {\n        const [x1, y1, x2, y2] = bbox.asIntXyxy();\n        const cx1 = Math.max(0, x1);\n        const cy1 = Math.max(0, y1);\n        const cx2 = Math.min(original.width, x2);\n        const cy2 = Math.min(original.height, y2);\n\n        let segmentedImage: RGBImage;\n        let finalMask = mask;\n        if (cx2 > cx1 && cy2 > cy1 && mask.data.length > 0) {\n            const cropW = cx2 - cx1;\n            const cropH = cy2 - cy1;\n            const mw = Math.min(mask.width, cropW);\n            const mh = Math.min(mask.height, cropH);\n            const segData = new Uint8Array(mw * mh * 3);\n            for (let row = 0; row < mh; row++) {\n                const srcRowOffset = ((cy1 + row) * original.width + cx1) * 3;\n                const dstRowOffset = row * mw * 3;\n                const maskRowOffset = row * mask.width;\n                for (let col = 0; col < mw; col++) {\n                    const m = mask.data[maskRowOffset + col];\n                    if (m !== 0) {\n                        const s = srcRowOffset + col * 3;\n                        const d = dstRowOffset + col * 3;\n                        segData[d] = original.data[s];\n                        segData[d + 1] = original.data[s + 1];\n                        segData[d + 2] = original.data[s + 2];\n                    }\n                }\n            }\n            segmentedImage = new RGBImage(segData, mw, mh);\n            if (mw !== mask.width || mh !== mask.height) {\n                const trimmed = new Uint8Array(mw * mh);\n                for (let row = 0; row < mh; row++) {\n                    trimmed.set(\n                        mask.data.subarray(row * mask.width, row * mask.width + mw),\n                        row * mw,\n                    );\n                }\n                finalMask = new Mask(trimmed, mw, mh);\n            }\n        } else {\n            finalMask = new Mask(new Uint8Array(0), 0, 0);\n            segmentedImage = new RGBImage(new Uint8Array(0), 0, 0);\n        }\n\n        const className = this._names[classId] ?? `class_${classId}`;\n\n        return {\n            classId,\n            className,\n            confidence,\n            bbox,\n            cls: classId,\n            name: className,\n            conf: confidence,\n            box: bbox,\n            mask: finalMask,\n            segmentedImage,\n        };\n    }\n\n    private _buildBoxes(\n        detections: readonly SegmentationResult[],\n        origShape: readonly [number, number],\n    ): Boxes {\n        const n = detections.length;\n        const xyxy = new Float32Array(n * 4);\n        const cls = new Int32Array(n);\n        const conf = new Float32Array(n);\n        for (let i = 0; i < n; i++) {\n            const d = detections[i] as SegmentationResult;\n            xyxy[i * 4] = d.bbox.x1;\n            xyxy[i * 4 + 1] = d.bbox.y1;\n            xyxy[i * 4 + 2] = d.bbox.x2;\n            xyxy[i * 4 + 3] = d.bbox.y2;\n            cls[i] = d.classId;\n            conf[i] = d.confidence;\n        }\n        return new Boxes(xyxy, cls, conf, origShape);\n    }\n\n    private _buildMasks(\n        detections: readonly SegmentationResult[],\n        origShape: readonly [number, number],\n    ): Masks {\n        const xyxy = new Float32Array(detections.length * 4);\n        for (let i = 0; i < detections.length; i++) {\n            const d = detections[i] as SegmentationResult;\n            xyxy[i * 4] = d.bbox.x1;\n            xyxy[i * 4 + 1] = d.bbox.y1;\n            xyxy[i * 4 + 2] = d.bbox.x2;\n            xyxy[i * 4 + 3] = d.bbox.y2;\n        }\n        return new Masks(\n            detections.map((d) => d.mask),\n            xyxy,\n            origShape,\n        );\n    }\n}\n"],"mappings":"+PAoFA,IAAa,EAAb,MAAa,UAAkB,EAAA,UAAW,CAGjB,MACA,QACA,OACA,WACA,eACA,cACA,eACA,eATrB,YACI,EACA,EACA,EACA,EACA,EACA,EACA,EACA,EACA,EACF,CACE,MAAM,CAAO,EATI,KAAA,MAAA,EACA,KAAA,QAAA,EACA,KAAA,OAAA,EACA,KAAA,WAAA,EACA,KAAA,eAAA,EACA,KAAA,cAAA,EACA,KAAA,eAAA,EACA,KAAA,eAAA,CAGrB,CAGA,aAAa,OAAO,EAAoB,EAA4B,CAAC,EAAuB,CACxF,IAAM,EAAsB,EAAQ,MAAQ,WAC5C,GAAI,IAAS,WACT,MAAU,MAAM,+BAA+B,EAAK,0BAA0B,EAElF,IAAM,EAAU,MAAM,EAAA,WAAW,OAAO,EAAO,CAAO,EAChD,EAAS,EAAA,cAAc,EAAQ,QAAU,OAAQ,CACnD,WAAY,EAAQ,UACxB,CAAC,EACK,EAAgC,CAAC,EACvC,IAAK,IAAI,EAAI,EAAG,EAAI,EAAO,OAAQ,IAC/B,EAAM,GAAK,EAAO,GAEtB,OAAO,IAAI,EACP,EACA,EACA,EACA,EACA,EAAQ,WAAa,CAAC,IAAK,GAAG,EAC9B,EAAQ,eAAiB,IACzB,EAAQ,cAAgB,IACxB,EAAQ,eAAiB,IACzB,EAAQ,eAAiB,EAC7B,CACJ,CAGA,IAAI,MAAsB,CACtB,OAAO,KAAK,KAChB,CAGA,IAAI,QAA4B,CAC5B,OAAO,KAAK,OAChB,CAGA,IAAI,OAA0C,CAC1C,OAAO,KAAK,MAChB,CAGA,IAAI,YAAqB,CACrB,OAAO,KAAK,QAAQ,MACxB,CAGA,MAAM,KACF,EACA,EAAmC,CAAC,EACN,CAC9B,OAAO,KAAK,QAAQ,EAAO,CAAO,CACtC,CAGA,MAAM,QACF,EACA,EAAmC,CAAC,EACN,CAC9B,IAAM,EAAO,OAAO,GAAU,SAAW,EAAQ,KAC3C,EAAW,MAAM,EAAA,UAAU,CAAK,EAChC,CAAE,SAAQ,QAAO,UAAS,UAAW,KAAK,YAAY,CAAQ,EAC9D,EAAU,MAAM,KAAK,SAAS,IAAI,EAAG,KAAK,SAAS,WAAY,CAAO,CAAC,EAEvE,CAAE,YAAW,cAAe,KAAK,cAAc,CAAO,EAEtD,EAAa,EAAA,cACf,EAAU,KACV,EAAU,KACV,EAAW,KACX,EAAW,KACX,CACI,WAAY,KAAK,QAAQ,OACzB,WAAY,KAAK,WAAW,GAC5B,YAAa,KAAK,WAAW,GAC7B,cAAe,EAAS,MACxB,eAAgB,EAAS,OACzB,UACA,SACA,QACA,cAAe,EAAQ,eAAiB,KAAK,eAC7C,aAAc,EAAQ,cAAgB,KAAK,cAC3C,cAAe,KAAK,eACpB,cAAe,KAAK,cACxB,CACJ,EAUM,GAPF,EAAQ,UAAY,IAAA,GAKd,OAJO,CACH,IAAM,EAAU,IAAI,IAAI,EAAQ,OAAO,EACvC,OAAO,EAAW,OAAQ,GAAM,EAAQ,IAAI,EAAE,OAAO,CAAC,CAC1D,EAAA,CAAG,EACH,CAEiB,IAAK,GAC5B,KAAK,aAAa,EAAU,EAAE,KAAM,EAAE,QAAS,EAAE,WAAY,EAAE,IAAI,CACvE,EAEM,EAAkC,CAAC,EAAS,OAAQ,EAAS,KAAK,EACxE,MAAO,CACH,IAAI,EAAA,oBACA,KAAK,YAAY,EAAY,CAAI,EACjC,KAAK,YAAY,EAAY,CAAI,EACjC,EACA,KAAK,OACL,EACA,EACA,CACJ,CACJ,CACJ,CAEA,YAAoB,EAKlB,CACE,GAAM,CAAC,EAAI,GAAM,KAAK,WAChB,EAAK,EAAA,UAAU,EAAO,EAAI,CAAE,EAGlC,MAAO,CACH,OAAQ,EAAA,gBAFA,EAAA,MADA,EAAA,UAAU,EAAG,KACP,EAAK,EAAG,MAAM,MAAO,EAAG,MAAM,OAAQ,CAE5B,EAAK,CAAC,EAAG,EAAG,EAAG,MAAM,OAAQ,EAAG,MAAM,KAAK,CAAC,EACpE,MAAO,EAAG,MACV,QAAS,EAAG,QACZ,OAAQ,EAAG,MACf,CACJ,CAEA,cAAsB,EAGpB,CACE,IAAI,EACA,EACJ,IAAK,IAAM,KAAQ,KAAK,SAAS,YAAa,CAC1C,IAAM,EAAI,EAAQ,GACd,IAAM,IAAA,KACN,EAAE,KAAK,SAAW,GAAK,IAAc,IAAA,GACrC,EAAY,EACL,EAAE,KAAK,SAAW,GAAK,IAAe,IAAA,KAC7C,EAAa,GAErB,CACA,GAAI,IAAc,IAAA,IAAa,IAAe,IAAA,GAAW,CACrD,IAAM,EAAS,KAAK,SAAS,YAAY,IACpC,GAAM,GAAG,EAAE,IAAI,KAAK,UAAU,EAAQ,EAAE,EAAE,MAAQ,CAAC,CAAC,GACzD,EACA,MAAU,MACN,uDAAuD,EAAO,KAAK,IAAI,EAAE,GAC7E,CACJ,CACA,MAAO,CAAE,YAAW,YAAW,CACnC,CAEA,aACI,EACA,EACA,EACA,EACA,EACkB,CAClB,GAAM,CAAC,EAAI,EAAI,EAAI,GAAM,EAAK,UAAU,EAClC,EAAM,KAAK,IAAI,EAAG,CAAE,EACpB,EAAM,KAAK,IAAI,EAAG,CAAE,EACpB,EAAM,KAAK,IAAI,EAAS,MAAO,CAAE,EACjC,EAAM,KAAK,IAAI,EAAS,OAAQ,CAAE,EAEpC,EACA,EAAY,EAChB,GAAI,EAAM,GAAO,EAAM,GAAO,EAAK,KAAK,OAAS,EAAG,CAChD,IAAM,EAAQ,EAAM,EACd,EAAQ,EAAM,EACd,EAAK,KAAK,IAAI,EAAK,MAAO,CAAK,EAC/B,EAAK,KAAK,IAAI,EAAK,OAAQ,CAAK,EAChC,EAAU,IAAI,WAAW,EAAK,EAAK,CAAC,EAC1C,IAAK,IAAI,EAAM,EAAG,EAAM,EAAI,IAAO,CAC/B,IAAM,IAAiB,EAAM,GAAO,EAAS,MAAQ,GAAO,EACtD,EAAe,EAAM,EAAK,EAC1B,EAAgB,EAAM,EAAK,MACjC,IAAK,IAAI,EAAM,EAAG,EAAM,EAAI,IAExB,GADU,EAAK,KAAK,EAAgB,KAC1B,EAAG,CACT,IAAM,EAAI,EAAe,EAAM,EACzB,EAAI,EAAe,EAAM,EAC/B,EAAQ,GAAK,EAAS,KAAK,GAC3B,EAAQ,EAAI,GAAK,EAAS,KAAK,EAAI,GACnC,EAAQ,EAAI,GAAK,EAAS,KAAK,EAAI,EACvC,CAER,CAEA,GADA,EAAiB,IAAI,EAAA,SAAS,EAAS,EAAI,CAAE,EACzC,IAAO,EAAK,OAAS,IAAO,EAAK,OAAQ,CACzC,IAAM,EAAU,IAAI,WAAW,EAAK,CAAE,EACtC,IAAK,IAAI,EAAM,EAAG,EAAM,EAAI,IACxB,EAAQ,IACJ,EAAK,KAAK,SAAS,EAAM,EAAK,MAAO,EAAM,EAAK,MAAQ,CAAE,EAC1D,EAAM,CACV,EAEJ,EAAY,IAAI,EAAA,KAAK,EAAS,EAAI,CAAE,CACxC,CACJ,KACI,GAAY,IAAI,EAAA,KAAK,IAAI,WAAe,EAAG,CAAC,EAC5C,EAAiB,IAAI,EAAA,SAAS,IAAI,WAAe,EAAG,CAAC,EAGzD,IAAM,EAAY,KAAK,OAAO,IAAY,SAAS,IAEnD,MAAO,CACH,UACA,YACA,aACA,OACA,IAAK,EACL,KAAM,EACN,KAAM,EACN,IAAK,EACL,KAAM,EACN,gBACJ,CACJ,CAEA,YACI,EACA,EACK,CACL,IAAM,EAAI,EAAW,OACf,EAAO,IAAI,aAAa,EAAI,CAAC,EAC7B,EAAM,IAAI,WAAW,CAAC,EACtB,EAAO,IAAI,aAAa,CAAC,EAC/B,IAAK,IAAI,EAAI,EAAG,EAAI,EAAG,IAAK,CACxB,IAAM,EAAI,EAAW,GACrB,EAAK,EAAI,GAAK,EAAE,KAAK,GACrB,EAAK,EAAI,EAAI,GAAK,EAAE,KAAK,GACzB,EAAK,EAAI,EAAI,GAAK,EAAE,KAAK,GACzB,EAAK,EAAI,EAAI,GAAK,EAAE,KAAK,GACzB,EAAI,GAAK,EAAE,QACX,EAAK,GAAK,EAAE,UAChB,CACA,OAAO,IAAI,EAAA,MAAM,EAAM,EAAK,EAAM,CAAS,CAC/C,CAEA,YACI,EACA,EACK,CACL,IAAM,EAAO,IAAI,aAAa,EAAW,OAAS,CAAC,EACnD,IAAK,IAAI,EAAI,EAAG,EAAI,EAAW,OAAQ,IAAK,CACxC,IAAM,EAAI,EAAW,GACrB,EAAK,EAAI,GAAK,EAAE,KAAK,GACrB,EAAK,EAAI,EAAI,GAAK,EAAE,KAAK,GACzB,EAAK,EAAI,EAAI,GAAK,EAAE,KAAK,GACzB,EAAK,EAAI,EAAI,GAAK,EAAE,KAAK,EAC7B,CACA,OAAO,IAAI,EAAA,MACP,EAAW,IAAK,GAAM,EAAE,IAAI,EAC5B,EACA,CACJ,CACJ,CACJ"}