tempest-react-sdk 0.23.0 → 0.24.0

package/dist/vision/tasks/classifier.cjs

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

@@ -1 +1 @@
-{"version":3,"file":"classifier.cjs","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"],"names":["IMAGENET_MEAN","IMAGENET_STD","Classifier","VisionTask","session","_labels","_names","_inputSize","_mean","_std","_applySoftmax","model","options","OrtSession","labels","resolveLabels","names","image","path","original","loadImage","tensor","outputs","firstOutputName","raw","fullProbs","indices","values","topK","probabilities","i","id","className","top","result","orig","ClassificationResults","Probs","tw","th","resized","resize","normalized","normalize","chw","toCHW","toFloat32Tensor","softmax"],"mappings":"uTAeMA,EAAmD,CAAC,KAAO,KAAO,IAAK,EACvEC,EAAkD,CAAC,KAAO,KAAO,IAAK,EAsDrE,MAAMC,UAAmBC,EAAAA,UAAW,CAC/B,YACJC,EACiBC,EACAC,EACAC,EACAC,EACAC,EACAC,EACnB,CACE,MAAMN,CAAO,EAPI,KAAA,QAAAC,EACA,KAAA,OAAAC,EACA,KAAA,WAAAC,EACA,KAAA,MAAAC,EACA,KAAA,KAAAC,EACA,KAAA,cAAAC,CAGrB,CARqB,QACA,OACA,WACA,MACA,KACA,cAMrB,aAAa,OAAOC,EAAoBC,EAAiD,CACrF,MAAMR,EAAU,MAAMS,EAAAA,WAAW,OAAOF,EAAOC,CAAO,EAChDE,EAASC,EAAAA,cAAcH,EAAQ,OAAQ,CAAE,WAAYA,EAAQ,WAAY,EACzEI,EAAgC,CAAA,EACtC,QAAS,EAAI,EAAG,EAAIF,EAAO,OAAQ,IAC/BE,EAAM,CAAC,EAAIF,EAAO,CAAC,EAEvB,OAAO,IAAIZ,EACPE,EACAU,EACAE,EACAJ,EAAQ,WAAa,CAAC,IAAK,GAAG,EAC9BA,EAAQ,MAAQZ,EAChBY,EAAQ,KAAOX,EACfW,EAAQ,cAAgB,EAAA,CAEhC,CAGA,IAAI,QAA4B,CAC5B,OAAO,KAAK,OAChB,CAGA,IAAI,OAA0C,CAC1C,OAAO,KAAK,MAChB,CAGA,IAAI,YAAqB,CACrB,OAAO,KAAK,QAAQ,MACxB,CAGA,MAAM,KACFK,EACAL,EAAoC,GACJ,CAChC,OAAO,KAAK,QAAQK,EAAOL,CAAO,CACtC,CAGA,MAAM,QACFK,EACAL,EAAoC,GACJ,CAChC,MAAMM,EAAO,OAAOD,GAAU,SAAWA,EAAQ,KAC3CE,EAAW,MAAMC,EAAAA,UAAUH,CAAK,EAChCI,EAAS,KAAK,YAAYF,CAAQ,EAClCG,EAAU,MAAM,KAAK,SAAS,IAAI,CAAE,CAAC,KAAK,SAAS,SAAS,EAAGD,EAAQ,EACvEE,EAAkB,KAAK,SAAS,YAAY,CAAC,EACnD,GAAIA,IAAoB,OACpB,MAAM,IAAI,MAAM,kCAAkC,EAEtD,MAAMC,EAAMF,EAAQC,CAAe,EACnC,GAAIC,IAAQ,OACR,MAAM,IAAI,MACN,2BAA2BD,CAAe,6BAAA,EAGlD,MAAME,EAAY,KAAK,aAAaD,EAAI,IAAoB,EAEtD,CAAE,QAAAE,EAAS,OAAAC,GAAWC,EAAAA,KAAKH,EAAWb,EAAQ,MAAQ,IAAI,EAC1DiB,EAAoC,CAAA,EAC1C,QAASC,EAAI,EAAGA,EAAIJ,EAAQ,OAAQI,IAAK,CACrC,MAAMC,EAAKL,EAAQI,CAAC,EACdE,EAAY,KAAK,QAAQD,CAAE,GAAK,SAASA,CAAE,GACjDF,EAAc,KAAK,CACf,QAASE,EACT,UAAAC,EACA,YAAaL,EAAOG,CAAC,EACrB,IAAKC,EACL,KAAMC,EACN,KAAML,EAAOG,CAAC,CAAA,CACjB,CACL,CACA,GAAID,EAAc,SAAW,EACzB,MAAM,IAAI,MAAM,sDAAsD,EAG1E,MAAMI,EAAMJ,EAAc,CAAC,EACrBK,EAA+B,CACjC,QAASD,EAAI,QACb,UAAWA,EAAI,UACf,WAAYA,EAAI,YAChB,IAAKA,EAAI,QACT,KAAMA,EAAI,UACV,KAAMA,EAAI,YACV,MAAOd,EACP,cAAAU,CAAA,EAGEM,EAAkC,CAAChB,EAAS,OAAQA,EAAS,KAAK,EACxE,MAAO,CACH,IAAIiB,EAAAA,sBACA,IAAIC,EAAAA,MAAMZ,CAAS,EACnBS,EACA,KAAK,OACLf,EACAgB,EACAjB,CAAA,CACJ,CAER,CAEQ,YAAYD,EAA6B,CAC7C,KAAM,CAACqB,EAAIC,CAAE,EAAI,KAAK,WAChBC,EAAUC,EAAAA,OAAOxB,EAAOqB,EAAIC,CAAE,EAC9BG,EAAaC,EAAAA,UAAUH,EAAS,KAAK,MAAO,KAAK,IAAI,EACrDI,EAAMC,EAAAA,MAAMH,EAAYF,EAAQ,MAAOA,EAAQ,OAAQ,CAAC,EAC9D,OAAOM,EAAAA,gBAAgBF,EAAK,CAAC,EAAG,EAAGJ,EAAQ,OAAQA,EAAQ,KAAK,CAAC,CACrE,CAEQ,aAAahB,EAAiC,CAClD,OAAO,KAAK,cAAgBuB,EAAAA,QAAQvB,CAAG,EAAI,IAAI,aAAaA,CAAG,CACnE,CACJ"}
+{"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

@@ -1,106 +1,90 @@
-import { OrtSession as g } from "../core/session.js";
-import { loadImage as y } from "../io/image.js";
-import { resolveLabels as N } from "../labels.js";
-import { topK as S, softmax as z } from "../postprocess/classification.js";
-import { resize as C, normalize as E, toCHW as x, toFloat32Tensor as I } from "../preprocess/image.js";
-import { ClassificationResults as T, Probs as v } from "../results.js";
-import { VisionTask as A } from "./base.js";
-const M = [0.485, 0.456, 0.406], F = [0.229, 0.224, 0.225];
-class d extends A {
-  constructor(e, t, r, s, i, o, n) {
-    super(e), this._labels = t, this._names = r, this._inputSize = s, this._mean = i, this._std = o, this._applySoftmax = n;
-  }
-  _labels;
-  _names;
-  _inputSize;
-  _mean;
-  _std;
-  _applySoftmax;
-  /** Load the model and resolve labels. */
-  static async create(e, t) {
-    const r = await g.create(e, t), s = N(t.labels, { numClasses: t.numClasses }), i = {};
-    for (let o = 0; o < s.length; o++)
-      i[o] = s[o];
-    return new d(
-      r,
-      s,
-      i,
-      t.inputSize ?? [224, 224],
-      t.mean ?? M,
-      t.std ?? F,
-      t.applySoftmax ?? !0
-    );
-  }
-  /** Class labels indexed by class id. */
-  get labels() {
-    return this._labels;
-  }
-  /** Class id → class name dict (matches Ultralytics' `model.names`). */
-  get names() {
-    return this._names;
-  }
-  /** Number of classes the model can predict. */
-  get numClasses() {
-    return this._labels.length;
-  }
-  /** Alias for {@link predict} (parity with PyTorch `nn.Module.__call__`). */
-  async call(e, t = {}) {
-    return this.predict(e, t);
-  }
-  /** Run classification on a single image. */
-  async predict(e, t = {}) {
-    const r = typeof e == "string" ? e : null, s = await y(e), i = this._preprocess(s), o = await this._session.run({ [this._session.inputName]: i }), n = this._session.outputNames[0];
-    if (n === void 0)
-      throw new Error("Classifier model has no outputs.");
-    const p = o[n];
-    if (p === void 0)
-      throw new Error(
-        `Classifier model output ${n} missing from run() result.`
-      );
-    const h = this._postprocess(p.data), { indices: u, values: _ } = S(h, t.topK ?? null), c = [];
-    for (let l = 0; l < u.length; l++) {
-      const m = u[l], f = this._labels[m] ?? `class_${m}`;
-      c.push({
-        classId: m,
-        className: f,
-        probability: _[l],
-        cls: m,
-        name: f,
-        conf: _[l]
-      });
-    }
-    if (c.length === 0)
-      throw new Error("Classifier produced no probabilities (empty output).");
-    const a = c[0], b = {
-      classId: a.classId,
-      className: a.className,
-      confidence: a.probability,
-      cls: a.classId,
-      name: a.className,
-      conf: a.probability,
-      image: s,
-      probabilities: c
-    }, w = [s.height, s.width];
-    return [
-      new T(
-        new v(h),
-        b,
-        this._names,
-        s,
-        w,
-        r
-      )
-    ];
-  }
-  _preprocess(e) {
-    const [t, r] = this._inputSize, s = C(e, t, r), i = E(s, this._mean, this._std), o = x(i, s.width, s.height, 3);
-    return I(o, [1, 3, s.height, s.width]);
-  }
-  _postprocess(e) {
-    return this._applySoftmax ? z(e) : new Float32Array(e);
-  }
-}
-export {
-  d as Classifier
+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);
+	}
 };
-//# sourceMappingURL=classifier.js.map
+//#endregion
+export { m as Classifier };
+
+//# sourceMappingURL=classifier.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"classifier.js","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"],"names":["IMAGENET_MEAN","IMAGENET_STD","Classifier","VisionTask","session","_labels","_names","_inputSize","_mean","_std","_applySoftmax","model","options","OrtSession","labels","resolveLabels","names","i","image","path","original","loadImage","tensor","outputs","firstOutputName","raw","fullProbs","indices","values","topK","probabilities","id","className","top","result","orig","ClassificationResults","Probs","tw","th","resized","resize","normalized","normalize","chw","toCHW","toFloat32Tensor","softmax"],"mappings":";;;;;;;AAeA,MAAMA,IAAmD,CAAC,OAAO,OAAO,KAAK,GACvEC,IAAkD,CAAC,OAAO,OAAO,KAAK;AAsDrE,MAAMC,UAAmBC,EAAW;AAAA,EAC/B,YACJC,GACiBC,GACAC,GACAC,GACAC,GACAC,GACAC,GACnB;AACE,UAAMN,CAAO,GAPI,KAAA,UAAAC,GACA,KAAA,SAAAC,GACA,KAAA,aAAAC,GACA,KAAA,QAAAC,GACA,KAAA,OAAAC,GACA,KAAA,gBAAAC;AAAA,EAGrB;AAAA,EARqB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EAMrB,aAAa,OAAOC,GAAoBC,GAAiD;AACrF,UAAMR,IAAU,MAAMS,EAAW,OAAOF,GAAOC,CAAO,GAChDE,IAASC,EAAcH,EAAQ,QAAQ,EAAE,YAAYA,EAAQ,YAAY,GACzEI,IAAgC,CAAA;AACtC,aAASC,IAAI,GAAGA,IAAIH,EAAO,QAAQG;AAC/B,MAAAD,EAAMC,CAAC,IAAIH,EAAOG,CAAC;AAEvB,WAAO,IAAIf;AAAA,MACPE;AAAA,MACAU;AAAA,MACAE;AAAA,MACAJ,EAAQ,aAAa,CAAC,KAAK,GAAG;AAAA,MAC9BA,EAAQ,QAAQZ;AAAA,MAChBY,EAAQ,OAAOX;AAAA,MACfW,EAAQ,gBAAgB;AAAA,IAAA;AAAA,EAEhC;AAAA;AAAA,EAGA,IAAI,SAA4B;AAC5B,WAAO,KAAK;AAAA,EAChB;AAAA;AAAA,EAGA,IAAI,QAA0C;AAC1C,WAAO,KAAK;AAAA,EAChB;AAAA;AAAA,EAGA,IAAI,aAAqB;AACrB,WAAO,KAAK,QAAQ;AAAA,EACxB;AAAA;AAAA,EAGA,MAAM,KACFM,GACAN,IAAoC,IACJ;AAChC,WAAO,KAAK,QAAQM,GAAON,CAAO;AAAA,EACtC;AAAA;AAAA,EAGA,MAAM,QACFM,GACAN,IAAoC,IACJ;AAChC,UAAMO,IAAO,OAAOD,KAAU,WAAWA,IAAQ,MAC3CE,IAAW,MAAMC,EAAUH,CAAK,GAChCI,IAAS,KAAK,YAAYF,CAAQ,GAClCG,IAAU,MAAM,KAAK,SAAS,IAAI,EAAE,CAAC,KAAK,SAAS,SAAS,GAAGD,GAAQ,GACvEE,IAAkB,KAAK,SAAS,YAAY,CAAC;AACnD,QAAIA,MAAoB;AACpB,YAAM,IAAI,MAAM,kCAAkC;AAEtD,UAAMC,IAAMF,EAAQC,CAAe;AACnC,QAAIC,MAAQ;AACR,YAAM,IAAI;AAAA,QACN,2BAA2BD,CAAe;AAAA,MAAA;AAGlD,UAAME,IAAY,KAAK,aAAaD,EAAI,IAAoB,GAEtD,EAAE,SAAAE,GAAS,QAAAC,MAAWC,EAAKH,GAAWd,EAAQ,QAAQ,IAAI,GAC1DkB,IAAoC,CAAA;AAC1C,aAASb,IAAI,GAAGA,IAAIU,EAAQ,QAAQV,KAAK;AACrC,YAAMc,IAAKJ,EAAQV,CAAC,GACde,IAAY,KAAK,QAAQD,CAAE,KAAK,SAASA,CAAE;AACjD,MAAAD,EAAc,KAAK;AAAA,QACf,SAASC;AAAA,QACT,WAAAC;AAAA,QACA,aAAaJ,EAAOX,CAAC;AAAA,QACrB,KAAKc;AAAA,QACL,MAAMC;AAAA,QACN,MAAMJ,EAAOX,CAAC;AAAA,MAAA,CACjB;AAAA,IACL;AACA,QAAIa,EAAc,WAAW;AACzB,YAAM,IAAI,MAAM,sDAAsD;AAG1E,UAAMG,IAAMH,EAAc,CAAC,GACrBI,IAA+B;AAAA,MACjC,SAASD,EAAI;AAAA,MACb,WAAWA,EAAI;AAAA,MACf,YAAYA,EAAI;AAAA,MAChB,KAAKA,EAAI;AAAA,MACT,MAAMA,EAAI;AAAA,MACV,MAAMA,EAAI;AAAA,MACV,OAAOb;AAAA,MACP,eAAAU;AAAA,IAAA,GAGEK,IAAkC,CAACf,EAAS,QAAQA,EAAS,KAAK;AACxE,WAAO;AAAA,MACH,IAAIgB;AAAA,QACA,IAAIC,EAAMX,CAAS;AAAA,QACnBQ;AAAA,QACA,KAAK;AAAA,QACLd;AAAA,QACAe;AAAA,QACAhB;AAAA,MAAA;AAAA,IACJ;AAAA,EAER;AAAA,EAEQ,YAAYD,GAA6B;AAC7C,UAAM,CAACoB,GAAIC,CAAE,IAAI,KAAK,YAChBC,IAAUC,EAAOvB,GAAOoB,GAAIC,CAAE,GAC9BG,IAAaC,EAAUH,GAAS,KAAK,OAAO,KAAK,IAAI,GACrDI,IAAMC,EAAMH,GAAYF,EAAQ,OAAOA,EAAQ,QAAQ,CAAC;AAC9D,WAAOM,EAAgBF,GAAK,CAAC,GAAG,GAAGJ,EAAQ,QAAQA,EAAQ,KAAK,CAAC;AAAA,EACrE;AAAA,EAEQ,aAAaf,GAAiC;AAClD,WAAO,KAAK,gBAAgBsB,EAAQtB,CAAG,IAAI,IAAI,aAAaA,CAAG;AAAA,EACnE;AACJ;"}
+{"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

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

@@ -1 +1 @@
-{"version":3,"file":"detector.cjs","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"],"names":["Detector","VisionTask","session","_head","_labels","_names","_inputSize","_confThreshold","_iouThreshold","_maxDetections","model","options","head","OrtSession","labels","resolveLabels","names","i","image","path","original","loadImage","tensor","scale","padLeft","padTop","outputs","firstOutputName","raw","decodedAll","decodeYolo","detections","allowed","d","orig","DetectionResults","tw","th","lb","letterbox","f32","toFloat32","chw","toCHW","toFloat32Tensor","bbox","classId","confidence","x1","y1","x2","y2","cx1","cy1","cx2","cy2","cropped","cw","ch","out","row","srcOffset","RGBImage","className","origShape","xyxy","cls","conf","Boxes"],"mappings":"4UA8EO,MAAMA,UAAiBC,EAAAA,UAAW,CAC7B,YACJC,EACiBC,EACAC,EACAC,EACAC,EACAC,EACAC,EACAC,EACnB,CACE,MAAMP,CAAO,EARI,KAAA,MAAAC,EACA,KAAA,QAAAC,EACA,KAAA,OAAAC,EACA,KAAA,WAAAC,EACA,KAAA,eAAAC,EACA,KAAA,cAAAC,EACA,KAAA,eAAAC,CAGrB,CATqB,MACA,QACA,OACA,WACA,eACA,cACA,eAMrB,aAAa,OAAOC,EAAoBC,EAA2B,GAAuB,CACtF,MAAMC,EAAqBD,EAAQ,MAAQ,OAC3C,GAAIC,IAAS,OACT,MAAM,IAAI,MAAM,8BAA8BA,CAAI,uBAAuB,EAE7E,MAAMV,EAAU,MAAMW,EAAAA,WAAW,OAAOH,EAAOC,CAAO,EAChDG,EAASC,EAAAA,cAAcJ,EAAQ,QAAU,OAAQ,CACnD,WAAYA,EAAQ,UAAA,CACvB,EACKK,EAAgC,CAAA,EACtC,QAASC,EAAI,EAAGA,EAAIH,EAAO,OAAQG,IAC/BD,EAAMC,CAAC,EAAIH,EAAOG,CAAC,EAEvB,OAAO,IAAIjB,EACPE,EACAU,EACAE,EACAE,EACAL,EAAQ,WAAa,CAAC,IAAK,GAAG,EAC9BA,EAAQ,eAAiB,IACzBA,EAAQ,cAAgB,IACxBA,EAAQ,eAAiB,GAAA,CAEjC,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,KACFO,EACAP,EAAkC,GACP,CAC3B,OAAO,KAAK,QAAQO,EAAOP,CAAO,CACtC,CAGA,MAAM,QACFO,EACAP,EAAkC,GACP,CAC3B,MAAMQ,EAAO,OAAOD,GAAU,SAAWA,EAAQ,KAC3CE,EAAW,MAAMC,EAAAA,UAAUH,CAAK,EAChC,CAAE,OAAAI,EAAQ,MAAAC,EAAO,QAAAC,EAAS,OAAAC,GAAW,KAAK,YAAYL,CAAQ,EAC9DM,EAAU,MAAM,KAAK,SAAS,IAAI,CAAE,CAAC,KAAK,SAAS,SAAS,EAAGJ,EAAQ,EAEvEK,EAAkB,KAAK,SAAS,YAAY,CAAC,EACnD,GAAIA,IAAoB,OACpB,MAAM,IAAI,MAAM,gCAAgC,EAEpD,MAAMC,EAAMF,EAAQC,CAAe,EACnC,GAAIC,IAAQ,OACR,MAAM,IAAI,MAAM,yBAAyBD,CAAe,6BAA6B,EAGzF,MAAME,EAAaC,EAAAA,WAAWF,EAAI,KAAsBA,EAAI,KAAM,CAC9D,cAAeR,EAAS,MACxB,eAAgBA,EAAS,OACzB,QAAAI,EACA,OAAAC,EACA,MAAAF,EACA,cAAeZ,EAAQ,eAAiB,KAAK,eAC7C,aAAcA,EAAQ,cAAgB,KAAK,cAC3C,cAAe,KAAK,cAAA,CACvB,EAUKoB,GAPFpB,EAAQ,UAAY,QACb,IAAM,CACH,MAAMqB,EAAU,IAAI,IAAIrB,EAAQ,OAAO,EACvC,OAAOkB,EAAW,OAAQI,GAAMD,EAAQ,IAAIC,EAAE,OAAO,CAAC,CAC1D,KACAJ,GAEiB,IAAKI,GAC5B,KAAK,aAAab,EAAUa,EAAE,KAAMA,EAAE,QAASA,EAAE,UAAU,CAAA,EAGzDC,EAAkC,CAACd,EAAS,OAAQA,EAAS,KAAK,EACxE,MAAO,CACH,IAAIe,EAAAA,iBACA,KAAK,YAAYJ,EAAYG,CAAI,EACjCH,EACA,KAAK,OACLX,EACAc,EACAf,CAAA,CACJ,CAER,CAEQ,YAAYD,EAKlB,CACE,KAAM,CAACkB,EAAIC,CAAE,EAAI,KAAK,WAChBC,EAAKC,EAAAA,UAAUrB,EAAOkB,EAAIC,CAAE,EAC5BG,EAAMC,EAAAA,UAAUH,EAAG,KAAK,EACxBI,EAAMC,EAAAA,MAAMH,EAAKF,EAAG,MAAM,MAAOA,EAAG,MAAM,OAAQ,CAAC,EACzD,MAAO,CACH,OAAQM,EAAAA,gBAAgBF,EAAK,CAAC,EAAG,EAAGJ,EAAG,MAAM,OAAQA,EAAG,MAAM,KAAK,CAAC,EACpE,MAAOA,EAAG,MACV,QAASA,EAAG,QACZ,OAAQA,EAAG,MAAA,CAEnB,CAEQ,aACJlB,EACAyB,EACAC,EACAC,EACe,CACf,KAAM,CAACC,EAAIC,EAAIC,EAAIC,CAAE,EAAIN,EAAK,UAAA,EACxBO,EAAM,KAAK,IAAI,EAAGJ,CAAE,EACpBK,EAAM,KAAK,IAAI,EAAGJ,CAAE,EACpBK,EAAM,KAAK,IAAIlC,EAAS,MAAO8B,CAAE,EACjCK,EAAM,KAAK,IAAInC,EAAS,OAAQ+B,CAAE,EAExC,IAAIK,EACJ,GAAIF,EAAMF,GAAOG,EAAMF,EAAK,CACxB,MAAMI,EAAKH,EAAMF,EACXM,EAAKH,EAAMF,EACXM,EAAM,IAAI,WAAWF,EAAKC,EAAK,CAAC,EACtC,QAASE,EAAM,EAAGA,EAAMF,EAAIE,IAAO,CAC/B,MAAMC,IAAcR,EAAMO,GAAOxC,EAAS,MAAQgC,GAAO,EACzDO,EAAI,IAAIvC,EAAS,KAAK,SAASyC,EAAWA,EAAYJ,EAAK,CAAC,EAAGG,EAAMH,EAAK,CAAC,CAC/E,CACAD,EAAU,IAAIM,EAAAA,SAASH,EAAKF,EAAIC,CAAE,CACtC,MACIF,EAAU,IAAIM,EAAAA,SAAS,IAAI,WAAW,CAAC,EAAG,EAAG,CAAC,EAGlD,MAAMC,EAAY,KAAK,OAAOjB,CAAO,GAAK,SAASA,CAAO,GAE1D,MAAO,CACH,QAAAA,EACA,UAAAiB,EACA,WAAAhB,EACA,KAAAF,EACA,IAAKC,EACL,KAAMiB,EACN,KAAMhB,EACN,IAAKF,EACL,aAAcW,CAAA,CAEtB,CAEQ,YACJzB,EACAiC,EACK,CACL,MAAM,EAAIjC,EAAW,OACfkC,EAAO,IAAI,aAAa,EAAI,CAAC,EAC7BC,EAAM,IAAI,WAAW,CAAC,EACtBC,EAAO,IAAI,aAAa,CAAC,EAC/B,QAASlD,EAAI,EAAGA,EAAI,EAAGA,IAAK,CACxB,MAAMgB,EAAIF,EAAWd,CAAC,EACtBgD,EAAKhD,EAAI,CAAC,EAAIgB,EAAE,KAAK,GACrBgC,EAAKhD,EAAI,EAAI,CAAC,EAAIgB,EAAE,KAAK,GACzBgC,EAAKhD,EAAI,EAAI,CAAC,EAAIgB,EAAE,KAAK,GACzBgC,EAAKhD,EAAI,EAAI,CAAC,EAAIgB,EAAE,KAAK,GACzBiC,EAAIjD,CAAC,EAAIgB,EAAE,QACXkC,EAAKlD,CAAC,EAAIgB,EAAE,UAChB,CACA,OAAO,IAAImC,EAAAA,MAAMH,EAAMC,EAAKC,EAAMH,CAAS,CAC/C,CACJ"}
+{"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"}