@mauriciobenjamin700/ort-vision-sdk-web 0.2.1

package/dist/index.js

@@ -0,0 +1,21 @@
+/**
+ * @ort-vision-sdk/web — high-level TypeScript SDK for browser computer
+ * vision inference with ONNX Runtime Web.
+ */
+export { BoundingBox, Mask, RGBImage, } from "./types.js";
+export { Boxes, ClassificationResults, DetectionResults, Masks, Probs, SegmentationResults, } from "./results.js";
+export { COCO_CLASSES, resolveLabels, } from "./labels.js";
+export { ImageLoadError, InferenceError, LabelMapError, ModelLoadError, OrtVisionError, ProviderNotAvailableError, } from "./core/exceptions.js";
+export { OrtSession, } from "./core/session.js";
+export { DEFAULT_PROVIDERS, resolveProviders } from "./core/providers.js";
+export { loadImage } from "./io/image.js";
+export { fromCv2, letterbox, normalize, resize, toCHW, toCv2, toFloat32, toFloat32Tensor, toTensor, } from "./preprocess/image.js";
+export { softmax, topK, } from "./postprocess/classification.js";
+export { batchedNms, decodeYolo, decodeYoloAnchors, decodeYoloV8, decodeYoloV8Anchors, nms, } from "./postprocess/detection.js";
+export { decodeYoloSeg, decodeYoloV8Seg, } from "./postprocess/segmentation.js";
+export { VisionTask } from "./tasks/base.js";
+export { Classifier, } from "./tasks/classifier.js";
+export { Detector, } from "./tasks/detector.js";
+export { Segmenter, } from "./tasks/segmenter.js";
+export const VERSION = "0.2.1";
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EACL,WAAW,EACX,IAAI,EACJ,QAAQ,GAKT,MAAM,YAAY,CAAC;AAEpB,OAAO,EACL,KAAK,EACL,qBAAqB,EACrB,gBAAgB,EAChB,KAAK,EACL,KAAK,EACL,mBAAmB,GACpB,MAAM,cAAc,CAAC;AAEtB,OAAO,EACL,YAAY,EAGZ,aAAa,GACd,MAAM,aAAa,CAAC;AAErB,OAAO,EACL,cAAc,EACd,cAAc,EACd,aAAa,EACb,cAAc,EACd,cAAc,EACd,yBAAyB,GAC1B,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAGL,UAAU,GACX,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAE1E,OAAO,EAAmB,SAAS,EAAE,MAAM,eAAe,CAAC;AAE3D,OAAO,EAEL,OAAO,EACP,SAAS,EACT,SAAS,EACT,MAAM,EACN,KAAK,EACL,KAAK,EACL,SAAS,EACT,eAAe,EACf,QAAQ,GACT,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EAEL,OAAO,EACP,IAAI,GACL,MAAM,iCAAiC,CAAC;AAEzC,OAAO,EAOL,UAAU,EACV,UAAU,EACV,iBAAiB,EACjB,YAAY,EACZ,mBAAmB,EACnB,GAAG,GACJ,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAIL,aAAa,EACb,eAAe,GAChB,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAGL,UAAU,GACX,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAIL,QAAQ,GACT,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAIL,SAAS,GACV,MAAM,sBAAsB,CAAC;AAE9B,MAAM,CAAC,MAAM,OAAO,GAAW,OAAO,CAAC"}

package/dist/io/image.d.ts

@@ -0,0 +1,17 @@
+/**
+ * Image loading utilities — convert any supported source into the canonical
+ * {@link RGBImage} format.
+ *
+ * Browser-only: uses `fetch`, `createImageBitmap`, and a Canvas 2D context to
+ * decode arbitrary inputs into a tightly-packed `Uint8Array` of HWC RGB pixels.
+ */
+import { RGBImage } from "../types.js";
+/** Anything {@link loadImage} accepts as an image input. */
+export type ImageInput = string | Blob | HTMLImageElement | HTMLCanvasElement | OffscreenCanvas | ImageBitmap | ImageData | RGBImage;
+/**
+ * Load an image from any supported source into a HWC uint8 RGB array.
+ *
+ * @throws {@link ImageLoadError} if the source cannot be decoded or has an unsupported shape.
+ */
+export declare function loadImage(source: ImageInput): Promise<RGBImage>;
+//# sourceMappingURL=image.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"image.d.ts","sourceRoot":"","sources":["../../src/io/image.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAQH,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAEvC,4DAA4D;AAC5D,MAAM,MAAM,UAAU,GAClB,MAAM,GACN,IAAI,GACJ,gBAAgB,GAChB,iBAAiB,GACjB,eAAe,GACf,WAAW,GACX,SAAS,GACT,QAAQ,CAAC;AAEb;;;;GAIG;AACH,wBAAsB,SAAS,CAAC,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,QAAQ,CAAC,CAqCrE"}

package/dist/io/image.js

@@ -0,0 +1,104 @@
+/**
+ * Image loading utilities — convert any supported source into the canonical
+ * {@link RGBImage} format.
+ *
+ * Browser-only: uses `fetch`, `createImageBitmap`, and a Canvas 2D context to
+ * decode arbitrary inputs into a tightly-packed `Uint8Array` of HWC RGB pixels.
+ */
+import { createCanvas, get2DContext, imageDataToRGB, } from "../core/canvas.js";
+import { ImageLoadError } from "../core/exceptions.js";
+import { RGBImage } from "../types.js";
+/**
+ * Load an image from any supported source into a HWC uint8 RGB array.
+ *
+ * @throws {@link ImageLoadError} if the source cannot be decoded or has an unsupported shape.
+ */
+export async function loadImage(source) {
+    if (source instanceof RGBImage) {
+        return source;
+    }
+    if (typeof ImageData !== "undefined" && source instanceof ImageData) {
+        return imageDataToRGB(source);
+    }
+    if (typeof source === "string") {
+        return loadFromUrl(source);
+    }
+    if (typeof Blob !== "undefined" && source instanceof Blob) {
+        return loadFromBlob(source);
+    }
+    if (typeof HTMLImageElement !== "undefined" && source instanceof HTMLImageElement) {
+        await waitForImageElement(source);
+        return drawableToRGB(source, source.naturalWidth, source.naturalHeight);
+    }
+    if (typeof HTMLCanvasElement !== "undefined" && source instanceof HTMLCanvasElement) {
+        return drawableToRGB(source, source.width, source.height);
+    }
+    if (typeof OffscreenCanvas !== "undefined" && source instanceof OffscreenCanvas) {
+        return drawableToRGB(source, source.width, source.height);
+    }
+    if (typeof ImageBitmap !== "undefined" && source instanceof ImageBitmap) {
+        return drawableToRGB(source, source.width, source.height);
+    }
+    throw new ImageLoadError(`Unsupported image source type: ${Object.prototype.toString.call(source)}`);
+}
+async function loadFromUrl(url) {
+    let response;
+    try {
+        response = await fetch(url);
+    }
+    catch (err) {
+        throw new ImageLoadError(`Failed to fetch image from ${url}: ${err.message}`, { cause: err });
+    }
+    if (!response.ok) {
+        throw new ImageLoadError(`Failed to fetch image from ${url}: HTTP ${response.status} ${response.statusText}`);
+    }
+    const blob = await response.blob();
+    return loadFromBlob(blob);
+}
+async function loadFromBlob(blob) {
+    let bitmap;
+    try {
+        bitmap = await createImageBitmap(blob);
+    }
+    catch (err) {
+        throw new ImageLoadError(`Failed to decode image blob: ${err.message}`, { cause: err });
+    }
+    try {
+        return drawableToRGB(bitmap, bitmap.width, bitmap.height);
+    }
+    finally {
+        bitmap.close();
+    }
+}
+function waitForImageElement(img) {
+    if (img.complete && img.naturalWidth > 0) {
+        return Promise.resolve();
+    }
+    return new Promise((resolve, reject) => {
+        const onLoad = () => {
+            cleanup();
+            resolve();
+        };
+        const onError = () => {
+            cleanup();
+            reject(new ImageLoadError("Failed to load HTMLImageElement (load event errored)"));
+        };
+        const cleanup = () => {
+            img.removeEventListener("load", onLoad);
+            img.removeEventListener("error", onError);
+        };
+        img.addEventListener("load", onLoad, { once: true });
+        img.addEventListener("error", onError, { once: true });
+    });
+}
+function drawableToRGB(drawable, width, height) {
+    if (width === 0 || height === 0) {
+        throw new ImageLoadError(`Cannot load image with zero dimension (${width}x${height}).`);
+    }
+    const canvas = createCanvas(width, height);
+    const ctx = get2DContext(canvas);
+    ctx.drawImage(drawable, 0, 0);
+    const imageData = ctx.getImageData(0, 0, width, height);
+    return imageDataToRGB(imageData);
+}
+//# sourceMappingURL=image.js.map

package/dist/io/image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image.js","sourceRoot":"","sources":["../../src/io/image.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EACL,YAAY,EACZ,YAAY,EACZ,cAAc,GACf,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAavC;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,MAAkB;IAChD,IAAI,MAAM,YAAY,QAAQ,EAAE,CAAC;QAC/B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,OAAO,SAAS,KAAK,WAAW,IAAI,MAAM,YAAY,SAAS,EAAE,CAAC;QACpE,OAAO,cAAc,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,WAAW,CAAC,MAAM,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,OAAO,IAAI,KAAK,WAAW,IAAI,MAAM,YAAY,IAAI,EAAE,CAAC;QAC1D,OAAO,YAAY,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IAED,IAAI,OAAO,gBAAgB,KAAK,WAAW,IAAI,MAAM,YAAY,gBAAgB,EAAE,CAAC;QAClF,MAAM,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAClC,OAAO,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;IAC1E,CAAC;IAED,IAAI,OAAO,iBAAiB,KAAK,WAAW,IAAI,MAAM,YAAY,iBAAiB,EAAE,CAAC;QACpF,OAAO,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,OAAO,eAAe,KAAK,WAAW,IAAI,MAAM,YAAY,eAAe,EAAE,CAAC;QAChF,OAAO,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,OAAO,WAAW,KAAK,WAAW,IAAI,MAAM,YAAY,WAAW,EAAE,CAAC;QACxE,OAAO,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC5D,CAAC;IAED,MAAM,IAAI,cAAc,CACtB,kCAAkC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAC3E,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,GAAW;IACpC,IAAI,QAAkB,CAAC;IACvB,IAAI,CAAC;QACH,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;IAC9B,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,IAAI,cAAc,CACtB,8BAA8B,GAAG,KAAM,GAAa,CAAC,OAAO,EAAE,EAC9D,EAAE,KAAK,EAAE,GAAG,EAAE,CACf,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,cAAc,CACtB,8BAA8B,GAAG,UAAU,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CACpF,CAAC;IACJ,CAAC;IACD,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IACnC,OAAO,YAAY,CAAC,IAAI,CAAC,CAAC;AAC5B,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,IAAU;IACpC,IAAI,MAAmB,CAAC;IACxB,IAAI,CAAC;QACH,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,CAAC;IACzC,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,IAAI,cAAc,CACtB,gCAAiC,GAAa,CAAC,OAAO,EAAE,EACxD,EAAE,KAAK,EAAE,GAAG,EAAE,CACf,CAAC;IACJ,CAAC;IACD,IAAI,CAAC;QACH,OAAO,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC5D,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,GAAqB;IAChD,IAAI,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,YAAY,GAAG,CAAC,EAAE,CAAC;QACzC,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IACD,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC3C,MAAM,MAAM,GAAG,GAAS,EAAE;YACxB,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC;QACF,MAAM,OAAO,GAAG,GAAS,EAAE;YACzB,OAAO,EAAE,CAAC;YACV,MAAM,CAAC,IAAI,cAAc,CAAC,sDAAsD,CAAC,CAAC,CAAC;QACrF,CAAC,CAAC;QACF,MAAM,OAAO,GAAG,GAAS,EAAE;YACzB,GAAG,CAAC,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACxC,GAAG,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC5C,CAAC,CAAC;QACF,GAAG,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,GAAG,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,aAAa,CACpB,QAA2B,EAC3B,KAAa,EACb,MAAc;IAEd,IAAI,KAAK,KAAK,CAAC,IAAI,MAAM,KAAK,CAAC,EAAE,CAAC;QAChC,MAAM,IAAI,cAAc,CAAC,0CAA0C,KAAK,IAAI,MAAM,IAAI,CAAC,CAAC;IAC1F,CAAC;IACD,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC3C,MAAM,GAAG,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;IACjC,GAAG,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9B,MAAM,SAAS,GAAG,GAAG,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;IACxD,OAAO,cAAc,CAAC,SAAS,CAAC,CAAC;AACnC,CAAC"}

package/dist/labels.d.ts

@@ -0,0 +1,38 @@
+/**
+ * Class label resolution: presets, lists, dicts, or auto-generated.
+ *
+ * Tasks call {@link resolveLabels} once at construction time to turn whatever
+ * the caller passed (preset name, array, dict, or `null`) into an ordered
+ * array of class names indexed by class id.
+ *
+ * In the browser there is no filesystem, so this module does not load labels
+ * from a path — fetch the file yourself and pass an array.
+ */
+/**
+ * Anything accepted by {@link resolveLabels}.
+ *
+ * - `string[]` / `readonly string[]`: explicit names indexed by class id.
+ * - `Record<number, string>`: sparse mapping (gaps filled with `class_<id>`).
+ * - `string`: a preset name (e.g. `"coco"`).
+ * - `null` / `undefined`: auto-generate `class_0` ... `class_{numClasses-1}`.
+ */
+export type LabelSpec = readonly string[] | Record<number, string> | string | null | undefined;
+/** COCO 2017 80-class labels in canonical class-id order. */
+export declare const COCO_CLASSES: readonly string[];
+export interface ResolveLabelsOptions {
+    /**
+     * Expected number of classes.
+     *
+     * - When `spec` is `null`/`undefined`, this is required to auto-generate names.
+     * - When `spec` is provided, it validates that the resolved length matches.
+     */
+    readonly numClasses?: number;
+}
+/**
+ * Resolve a labels specification into an ordered array of class names.
+ *
+ * @throws {@link LabelMapError} if the spec is invalid, the preset is unknown,
+ *   or the resolved length disagrees with `numClasses`.
+ */
+export declare function resolveLabels(spec: LabelSpec, options?: ResolveLabelsOptions): readonly string[];
+//# sourceMappingURL=labels.d.ts.map

package/dist/labels.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"labels.d.ts","sourceRoot":"","sources":["../src/labels.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAIH;;;;;;;GAOG;AACH,MAAM,MAAM,SAAS,GACjB,SAAS,MAAM,EAAE,GACjB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACtB,MAAM,GACN,IAAI,GACJ,SAAS,CAAC;AAEd,6DAA6D;AAC7D,eAAO,MAAM,YAAY,EAAE,SAAS,MAAM,EAaxC,CAAC;AAMH,MAAM,WAAW,oBAAoB;IACnC;;;;;OAKG;IACH,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC;CAC9B;AAED;;;;;GAKG;AACH,wBAAgB,aAAa,CAC3B,IAAI,EAAE,SAAS,EACf,OAAO,GAAE,oBAAyB,GACjC,SAAS,MAAM,EAAE,CAQnB"}

package/dist/labels.js

@@ -0,0 +1,71 @@
+/**
+ * Class label resolution: presets, lists, dicts, or auto-generated.
+ *
+ * Tasks call {@link resolveLabels} once at construction time to turn whatever
+ * the caller passed (preset name, array, dict, or `null`) into an ordered
+ * array of class names indexed by class id.
+ *
+ * In the browser there is no filesystem, so this module does not load labels
+ * from a path — fetch the file yourself and pass an array.
+ */
+import { LabelMapError } from "./core/exceptions.js";
+/** COCO 2017 80-class labels in canonical class-id order. */
+export const COCO_CLASSES = Object.freeze([
+    "person", "bicycle", "car", "motorcycle", "airplane", "bus", "train", "truck",
+    "boat", "traffic light", "fire hydrant", "stop sign", "parking meter", "bench",
+    "bird", "cat", "dog", "horse", "sheep", "cow", "elephant", "bear", "zebra",
+    "giraffe", "backpack", "umbrella", "handbag", "tie", "suitcase", "frisbee",
+    "skis", "snowboard", "sports ball", "kite", "baseball bat", "baseball glove",
+    "skateboard", "surfboard", "tennis racket", "bottle", "wine glass", "cup",
+    "fork", "knife", "spoon", "bowl", "banana", "apple", "sandwich", "orange",
+    "broccoli", "carrot", "hot dog", "pizza", "donut", "cake", "chair", "couch",
+    "potted plant", "bed", "dining table", "toilet", "tv", "laptop", "mouse",
+    "remote", "keyboard", "cell phone", "microwave", "oven", "toaster", "sink",
+    "refrigerator", "book", "clock", "vase", "scissors", "teddy bear", "hair drier",
+    "toothbrush",
+]);
+const PRESETS = {
+    coco: COCO_CLASSES,
+};
+/**
+ * Resolve a labels specification into an ordered array of class names.
+ *
+ * @throws {@link LabelMapError} if the spec is invalid, the preset is unknown,
+ *   or the resolved length disagrees with `numClasses`.
+ */
+export function resolveLabels(spec, options = {}) {
+    const labels = resolve(spec, options.numClasses);
+    if (options.numClasses !== undefined && labels.length !== options.numClasses) {
+        throw new LabelMapError(`Resolved ${labels.length} labels but the model has ${options.numClasses} classes.`);
+    }
+    return labels;
+}
+function resolve(spec, numClasses) {
+    if (spec === null || spec === undefined) {
+        if (numClasses === undefined) {
+            throw new LabelMapError("Cannot auto-generate labels without numClasses. Pass an explicit labels spec or numClasses.");
+        }
+        return Array.from({ length: numClasses }, (_, i) => `class_${i}`);
+    }
+    if (Array.isArray(spec)) {
+        return [...spec];
+    }
+    if (typeof spec === "string") {
+        const preset = PRESETS[spec];
+        if (preset !== undefined) {
+            return preset;
+        }
+        throw new LabelMapError(`Unknown labels preset: ${JSON.stringify(spec)}. Known presets: ${Object.keys(PRESETS).join(", ")}.`);
+    }
+    if (typeof spec === "object") {
+        const map = spec;
+        const ids = Object.keys(map).map((k) => Number(k));
+        if (ids.length === 0) {
+            return [];
+        }
+        const maxId = Math.max(...ids);
+        return Array.from({ length: maxId + 1 }, (_, i) => map[i] ?? `class_${i}`);
+    }
+    throw new LabelMapError(`Unsupported labels spec type: ${typeof spec}.`);
+}
+//# sourceMappingURL=labels.js.map

package/dist/labels.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"labels.js","sourceRoot":"","sources":["../src/labels.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAiBrD,6DAA6D;AAC7D,MAAM,CAAC,MAAM,YAAY,GAAsB,MAAM,CAAC,MAAM,CAAC;IAC3D,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO;IAC7E,MAAM,EAAE,eAAe,EAAE,cAAc,EAAE,WAAW,EAAE,eAAe,EAAE,OAAO;IAC9E,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO;IAC1E,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,SAAS;IAC1E,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,EAAE,cAAc,EAAE,gBAAgB;IAC5E,YAAY,EAAE,WAAW,EAAE,eAAe,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK;IACzE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ;IACzE,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO;IAC3E,cAAc,EAAE,KAAK,EAAE,cAAc,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO;IACxE,QAAQ,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;IAC1E,cAAc,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,YAAY;IAC/E,YAAY;CACb,CAAC,CAAC;AAEH,MAAM,OAAO,GAAgD;IAC3D,IAAI,EAAE,YAAY;CACnB,CAAC;AAYF;;;;;GAKG;AACH,MAAM,UAAU,aAAa,CAC3B,IAAe,EACf,UAAgC,EAAE;IAElC,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;IACjD,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,KAAK,OAAO,CAAC,UAAU,EAAE,CAAC;QAC7E,MAAM,IAAI,aAAa,CACrB,YAAY,MAAM,CAAC,MAAM,6BAA6B,OAAO,CAAC,UAAU,WAAW,CACpF,CAAC;IACJ,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,OAAO,CAAC,IAAe,EAAE,UAA8B;IAC9D,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACxC,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;YAC7B,MAAM,IAAI,aAAa,CACrB,6FAA6F,CAC9F,CAAC;QACJ,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACpE,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,OAAO,CAAC,GAAG,IAAI,CAAC,CAAC;IACnB,CAAC;IAED,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAC7B,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,MAAM,IAAI,aAAa,CACrB,0BAA0B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,oBAAoB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CACrG,CAAC;IACJ,CAAC;IAED,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,IAA8B,CAAC;QAC3C,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACnD,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrB,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC;QAC/B,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,EAAE,CAAC,CAAC;IAC7E,CAAC;IAED,MAAM,IAAI,aAAa,CAAC,iCAAiC,OAAO,IAAI,GAAG,CAAC,CAAC;AAC3E,CAAC"}

package/dist/postprocess/classification.d.ts

@@ -0,0 +1,16 @@
+/**
+ * Classification head postprocessing — softmax + top-k.
+ */
+/** Apply numerically-stable softmax to a 1-D vector of logits. */
+export declare function softmax(logits: Float32Array | readonly number[]): Float32Array;
+export interface TopKResult {
+    readonly indices: Int32Array;
+    readonly values: Float32Array;
+}
+/**
+ * Return the top-k entries of a 1-D probability vector, sorted descending.
+ *
+ * @param k Number of entries to return; `null` returns all entries.
+ */
+export declare function topK(probabilities: Float32Array, k: number | null): TopKResult;
+//# sourceMappingURL=classification.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"classification.d.ts","sourceRoot":"","sources":["../../src/postprocess/classification.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,kEAAkE;AAClE,wBAAgB,OAAO,CACrB,MAAM,EAAE,YAAY,GAAG,SAAS,MAAM,EAAE,GACvC,YAAY,CAkBd;AAED,MAAM,WAAW,UAAU;IACzB,QAAQ,CAAC,OAAO,EAAE,UAAU,CAAC;IAC7B,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAC;CAC/B;AAED;;;;GAIG;AACH,wBAAgB,IAAI,CAClB,aAAa,EAAE,YAAY,EAC3B,CAAC,EAAE,MAAM,GAAG,IAAI,GACf,UAAU,CAgBZ"}

package/dist/postprocess/classification.js

@@ -0,0 +1,46 @@
+/**
+ * Classification head postprocessing — softmax + top-k.
+ */
+/** Apply numerically-stable softmax to a 1-D vector of logits. */
+export function softmax(logits) {
+    const n = logits.length;
+    const out = new Float32Array(n);
+    let max = -Infinity;
+    for (let i = 0; i < n; i++) {
+        const v = logits[i];
+        if (v > max)
+            max = v;
+    }
+    let sum = 0;
+    for (let i = 0; i < n; i++) {
+        const e = Math.exp(logits[i] - max);
+        out[i] = e;
+        sum += e;
+    }
+    for (let i = 0; i < n; i++) {
+        out[i] = out[i] / sum;
+    }
+    return out;
+}
+/**
+ * Return the top-k entries of a 1-D probability vector, sorted descending.
+ *
+ * @param k Number of entries to return; `null` returns all entries.
+ */
+export function topK(probabilities, k) {
+    const n = probabilities.length;
+    const kEff = k === null ? n : Math.min(Math.max(0, k), n);
+    const idx = new Array(n);
+    for (let i = 0; i < n; i++)
+        idx[i] = i;
+    idx.sort((a, b) => probabilities[b] - probabilities[a]);
+    const indices = new Int32Array(kEff);
+    const values = new Float32Array(kEff);
+    for (let i = 0; i < kEff; i++) {
+        const j = idx[i];
+        indices[i] = j;
+        values[i] = probabilities[j];
+    }
+    return { indices, values };
+}
+//# sourceMappingURL=classification.js.map

package/dist/postprocess/classification.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"classification.js","sourceRoot":"","sources":["../../src/postprocess/classification.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,kEAAkE;AAClE,MAAM,UAAU,OAAO,CACrB,MAAwC;IAExC,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;IACxB,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC;IACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAW,CAAC;QAC9B,IAAI,CAAC,GAAG,GAAG;YAAE,GAAG,GAAG,CAAC,CAAC;IACvB,CAAC;IACD,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAE,MAAM,CAAC,CAAC,CAAY,GAAG,GAAG,CAAC,CAAC;QAChD,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACX,GAAG,IAAI,CAAC,CAAC;IACX,CAAC;IACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,GAAG,CAAC,CAAC,CAAC,GAAI,GAAG,CAAC,CAAC,CAAY,GAAG,GAAG,CAAC;IACpC,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAOD;;;;GAIG;AACH,MAAM,UAAU,IAAI,CAClB,aAA2B,EAC3B,CAAgB;IAEhB,MAAM,CAAC,GAAG,aAAa,CAAC,MAAM,CAAC;IAC/B,MAAM,IAAI,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAE1D,MAAM,GAAG,GAAG,IAAI,KAAK,CAAS,CAAC,CAAC,CAAC;IACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;QAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACvC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAE,aAAa,CAAC,CAAC,CAAY,GAAI,aAAa,CAAC,CAAC,CAAY,CAAC,CAAC;IAEhF,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,CAAW,CAAC;QAC3B,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACf,MAAM,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,CAAC,CAAW,CAAC;IACzC,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AAC7B,CAAC"}

package/dist/postprocess/detection.d.ts

@@ -0,0 +1,112 @@
+/**
+ * Detection head postprocessing: anchor-free YOLO decoding + non-maximum suppression.
+ *
+ * The shared {@link decodeYoloAnchors} helper does the per-anchor work that
+ * is identical for plain detection and segmentation (transpose, xywh→xyxy,
+ * letterbox unmap, per-class NMS, sort & cap). {@link decodeYolo} is a thin
+ * wrapper around it; the segmentation module ({@link ./segmentation.js})
+ * calls the helper directly so it can also recover the per-anchor mask
+ * coefficients.
+ *
+ * Works for any YOLO export with the post-v8 anchor-free head:
+ * **YOLOv8 / v9 / v10 / v11 / v12** detect heads, all of which share the
+ * `[1, 4 + nc, N]` output layout.
+ */
+import { BoundingBox } from "../types.js";
+/**
+ * Greedy non-maximum suppression on axis-aligned bounding boxes.
+ *
+ * Mirrors `torchvision.ops.nms` (keeps boxes with the highest score, drops
+ * any subsequent box whose IoU exceeds the threshold).
+ *
+ * @param boxes Flat array of length `4 * N` in xyxy order: `[x1,y1,x2,y2, ...]`.
+ * @param scores Detection score per box, length `N`.
+ * @param iouThreshold Boxes with IoU above this threshold relative to a kept box are suppressed.
+ * @returns Indices of kept boxes, in descending score order.
+ */
+export declare function nms(boxes: Float32Array, scores: Float32Array, iouThreshold: number): Int32Array;
+/**
+ * Per-class NMS — boxes are suppressed only by other boxes of the same class.
+ *
+ * Mirrors `torchvision.ops.batched_nms`.
+ *
+ * @param boxes Flat array of length `4 * N` in xyxy order.
+ * @param scores Detection score per box, length `N`.
+ * @param idxs Class index per box, length `N`. Boxes with different `idxs`
+ *   never suppress each other.
+ * @param iouThreshold IoU threshold for suppression within a class.
+ */
+export declare function batchedNms(boxes: Float32Array, scores: Float32Array, idxs: Int32Array, iouThreshold: number): Int32Array;
+export interface DecodeYoloAnchorsOptions {
+    /** Number of class-score channels following the 4 box channels. */
+    readonly numClasses: number;
+    readonly originalWidth: number;
+    readonly originalHeight: number;
+    readonly padLeft: number;
+    readonly padTop: number;
+    readonly scale: number;
+    readonly confThreshold: number;
+    readonly iouThreshold: number;
+    readonly maxDetections: number;
+}
+export interface DecodedAnchors {
+    /** Indices into the original `numAnchors` axis, in descending confidence order. */
+    readonly anchorIndices: Int32Array;
+    /** `[k, 4]` boxes in original-image pixel coords, flat row-major xyxy. */
+    readonly boxesXyxy: Float32Array;
+    /** Predicted class id per survivor. */
+    readonly classIds: Int32Array;
+    /** Confidence per survivor. */
+    readonly confidences: Float32Array;
+}
+/**
+ * Shared YOLO per-anchor decode used by both detection and segmentation
+ * (v8 / v9 / v10 / v11 / v12).
+ *
+ * Only the first `4 + numClasses` channels are read; later channels (e.g.
+ * mask coefficients) are ignored — callers can fetch them via the returned
+ * {@link DecodedAnchors.anchorIndices}.
+ *
+ * @param data Flat per-anchor output, length `channels * numAnchors`.
+ * @param dims Dims as reported by ORT, e.g. `[1, 84, 8400]` (det) or
+ *   `[1, 116, 8400]` (seg). The leading batch dim must be 1.
+ */
+export declare function decodeYoloAnchors(data: Float32Array, dims: readonly number[], options: DecodeYoloAnchorsOptions): DecodedAnchors;
+export interface DecodeYoloOptions {
+    readonly originalWidth: number;
+    readonly originalHeight: number;
+    readonly padLeft: number;
+    readonly padTop: number;
+    readonly scale: number;
+    readonly confThreshold: number;
+    readonly iouThreshold: number;
+    readonly maxDetections: number;
+}
+export interface DecodedDetection {
+    readonly bbox: BoundingBox;
+    readonly classId: number;
+    readonly confidence: number;
+}
+/**
+ * Decode an anchor-free YOLO detection output into a list of detections.
+ *
+ * Works for **YOLOv8 / v9 / v10 / v11 / v12** detect heads.
+ *
+ * Expected raw shape: `[1, 4 + numClasses, N]`. `numClasses` is inferred
+ * from the channel count.
+ */
+export declare function decodeYolo(output: Float32Array, outputDims: readonly number[], options: DecodeYoloOptions): DecodedDetection[];
+/**
+ * @deprecated since 0.2.0 — use {@link decodeYolo}. Same behavior; the
+ * decoder covers v8/v9/v10/v11/v12 detect heads. Will be removed in 0.3.0.
+ */
+export declare function decodeYoloV8(output: Float32Array, outputDims: readonly number[], options: DecodeYoloOptions): DecodedDetection[];
+/**
+ * @deprecated since 0.2.0 — use {@link decodeYoloAnchors}. Will be removed in 0.3.0.
+ */
+export declare function decodeYoloV8Anchors(data: Float32Array, dims: readonly number[], options: DecodeYoloAnchorsOptions): DecodedAnchors;
+/** @deprecated since 0.2.0 — use {@link DecodeYoloAnchorsOptions}. */
+export type DecodeYoloV8AnchorsOptions = DecodeYoloAnchorsOptions;
+/** @deprecated since 0.2.0 — use {@link DecodeYoloOptions}. */
+export type DecodeYoloV8Options = DecodeYoloOptions;
+//# sourceMappingURL=detection.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"detection.d.ts","sourceRoot":"","sources":["../../src/postprocess/detection.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C;;;;;;;;;;GAUG;AACH,wBAAgB,GAAG,CACjB,KAAK,EAAE,YAAY,EACnB,MAAM,EAAE,YAAY,EACpB,YAAY,EAAE,MAAM,GACnB,UAAU,CAsDZ;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,UAAU,CACxB,KAAK,EAAE,YAAY,EACnB,MAAM,EAAE,YAAY,EACpB,IAAI,EAAE,UAAU,EAChB,YAAY,EAAE,MAAM,GACnB,UAAU,CAgCZ;AAED,MAAM,WAAW,wBAAwB;IACvC,mEAAmE;IACnE,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,cAAc,EAAE,MAAM,CAAC;IAChC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;CAChC;AAED,MAAM,WAAW,cAAc;IAC7B,mFAAmF;IACnF,QAAQ,CAAC,aAAa,EAAE,UAAU,CAAC;IACnC,0EAA0E;IAC1E,QAAQ,CAAC,SAAS,EAAE,YAAY,CAAC;IACjC,uCAAuC;IACvC,QAAQ,CAAC,QAAQ,EAAE,UAAU,CAAC;IAC9B,+BAA+B;IAC/B,QAAQ,CAAC,WAAW,EAAE,YAAY,CAAC;CACpC;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,iBAAiB,CAC/B,IAAI,EAAE,YAAY,EAClB,IAAI,EAAE,SAAS,MAAM,EAAE,EACvB,OAAO,EAAE,wBAAwB,GAChC,cAAc,CAyHhB;AAWD,MAAM,WAAW,iBAAiB;IAChC,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,cAAc,EAAE,MAAM,CAAC;IAChC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;CAChC;AAED,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,CAAC,IAAI,EAAE,WAAW,CAAC;IAC3B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;CAC7B;AAED;;;;;;;GAOG;AACH,wBAAgB,UAAU,CACxB,MAAM,EAAE,YAAY,EACpB,UAAU,EAAE,SAAS,MAAM,EAAE,EAC7B,OAAO,EAAE,iBAAiB,GACzB,gBAAgB,EAAE,CA4BpB;AAKD;;;GAGG;AACH,wBAAgB,YAAY,CAC1B,MAAM,EAAE,YAAY,EACpB,UAAU,EAAE,SAAS,MAAM,EAAE,EAC7B,OAAO,EAAE,iBAAiB,GACzB,gBAAgB,EAAE,CASpB;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,IAAI,EAAE,YAAY,EAClB,IAAI,EAAE,SAAS,MAAM,EAAE,EACvB,OAAO,EAAE,wBAAwB,GAChC,cAAc,CAShB;AAED,sEAAsE;AACtE,MAAM,MAAM,0BAA0B,GAAG,wBAAwB,CAAC;AAElE,+DAA+D;AAC/D,MAAM,MAAM,mBAAmB,GAAG,iBAAiB,CAAC"}