@camstack/addon-vision 0.1.2 → 0.1.3

package/dist/chunk-2IOKI4ES.mjs

@@ -1,335 +0,0 @@
-// src/shared/node-engine.ts
-import * as path from "path";
-var BACKEND_TO_PROVIDER = {
-  cpu: "cpu",
-  coreml: "coreml",
-  cuda: "cuda",
-  tensorrt: "tensorrt",
-  dml: "dml"
-};
-var BACKEND_TO_DEVICE = {
-  cpu: "cpu",
-  coreml: "gpu-mps",
-  cuda: "gpu-cuda",
-  tensorrt: "tensorrt"
-};
-var NodeInferenceEngine = class {
-  constructor(modelPath, backend) {
-    this.modelPath = modelPath;
-    this.backend = backend;
-    this.device = BACKEND_TO_DEVICE[backend] ?? "cpu";
-  }
-  runtime = "onnx";
-  device;
-  session = null;
-  async initialize() {
-    const ort = await import("onnxruntime-node");
-    const provider = BACKEND_TO_PROVIDER[this.backend] ?? "cpu";
-    const absModelPath = path.isAbsolute(this.modelPath) ? this.modelPath : path.resolve(process.cwd(), this.modelPath);
-    const sessionOptions = {
-      executionProviders: [provider]
-    };
-    this.session = await ort.InferenceSession.create(absModelPath, sessionOptions);
-  }
-  async run(input, inputShape) {
-    if (!this.session) {
-      throw new Error("NodeInferenceEngine: not initialized \u2014 call initialize() first");
-    }
-    const ort = await import("onnxruntime-node");
-    const sess = this.session;
-    const inputName = sess.inputNames[0];
-    const tensor = new ort.Tensor("float32", input, [...inputShape]);
-    const feeds = { [inputName]: tensor };
-    const results = await sess.run(feeds);
-    const outputName = sess.outputNames[0];
-    const outputTensor = results[outputName];
-    return outputTensor.data;
-  }
-  async runMultiOutput(input, inputShape) {
-    if (!this.session) {
-      throw new Error("NodeInferenceEngine: not initialized \u2014 call initialize() first");
-    }
-    const ort = await import("onnxruntime-node");
-    const sess = this.session;
-    const inputName = sess.inputNames[0];
-    const tensor = new ort.Tensor("float32", input, [...inputShape]);
-    const feeds = { [inputName]: tensor };
-    const results = await sess.run(feeds);
-    const out = {};
-    for (const name of sess.outputNames) {
-      out[name] = results[name].data;
-    }
-    return out;
-  }
-  async dispose() {
-    this.session = null;
-  }
-};
-
-// src/shared/python-engine.ts
-import { spawn } from "child_process";
-var PythonInferenceEngine = class {
-  constructor(pythonPath, scriptPath, runtime, modelPath, extraArgs = []) {
-    this.pythonPath = pythonPath;
-    this.scriptPath = scriptPath;
-    this.modelPath = modelPath;
-    this.extraArgs = extraArgs;
-    this.runtime = runtime;
-    const runtimeDeviceMap = {
-      onnx: "cpu",
-      coreml: "gpu-mps",
-      pytorch: "cpu",
-      openvino: "cpu",
-      tflite: "cpu"
-    };
-    this.device = runtimeDeviceMap[runtime];
-  }
-  runtime;
-  device;
-  process = null;
-  receiveBuffer = Buffer.alloc(0);
-  pendingResolve = null;
-  pendingReject = null;
-  async initialize() {
-    const args = [this.scriptPath, this.modelPath, ...this.extraArgs];
-    this.process = spawn(this.pythonPath, args, {
-      stdio: ["pipe", "pipe", "pipe"]
-    });
-    if (!this.process.stdout || !this.process.stdin) {
-      throw new Error("PythonInferenceEngine: failed to create process pipes");
-    }
-    this.process.stderr?.on("data", (chunk) => {
-      process.stderr.write(`[python-engine] ${chunk.toString()}`);
-    });
-    this.process.on("error", (err) => {
-      this.pendingReject?.(err);
-      this.pendingReject = null;
-      this.pendingResolve = null;
-    });
-    this.process.on("exit", (code) => {
-      if (code !== 0) {
-        const err = new Error(`PythonInferenceEngine: process exited with code ${code}`);
-        this.pendingReject?.(err);
-        this.pendingReject = null;
-        this.pendingResolve = null;
-      }
-    });
-    this.process.stdout.on("data", (chunk) => {
-      this.receiveBuffer = Buffer.concat([this.receiveBuffer, chunk]);
-      this._tryReceive();
-    });
-    await new Promise((resolve2, reject) => {
-      const timeout = setTimeout(() => resolve2(), 2e3);
-      this.process?.on("error", (err) => {
-        clearTimeout(timeout);
-        reject(err);
-      });
-      this.process?.on("exit", (code) => {
-        clearTimeout(timeout);
-        if (code !== 0) {
-          reject(new Error(`PythonInferenceEngine: process exited early with code ${code}`));
-        }
-      });
-    });
-  }
-  _tryReceive() {
-    if (this.receiveBuffer.length < 4) return;
-    const length = this.receiveBuffer.readUInt32LE(0);
-    if (this.receiveBuffer.length < 4 + length) return;
-    const jsonBytes = this.receiveBuffer.subarray(4, 4 + length);
-    this.receiveBuffer = this.receiveBuffer.subarray(4 + length);
-    const resolve2 = this.pendingResolve;
-    const reject = this.pendingReject;
-    this.pendingResolve = null;
-    this.pendingReject = null;
-    if (!resolve2) return;
-    try {
-      const parsed = JSON.parse(jsonBytes.toString("utf8"));
-      resolve2(parsed);
-    } catch (err) {
-      reject?.(err instanceof Error ? err : new Error(String(err)));
-    }
-  }
-  /** Send JPEG buffer, receive JSON detection results */
-  async runJpeg(jpeg) {
-    if (!this.process?.stdin) {
-      throw new Error("PythonInferenceEngine: process not initialized");
-    }
-    return new Promise((resolve2, reject) => {
-      this.pendingResolve = resolve2;
-      this.pendingReject = reject;
-      const lengthBuf = Buffer.allocUnsafe(4);
-      lengthBuf.writeUInt32LE(jpeg.length, 0);
-      this.process.stdin.write(Buffer.concat([lengthBuf, jpeg]));
-    });
-  }
-  /** IInferenceEngine.run — wraps runJpeg for compatibility */
-  async run(_input, _inputShape) {
-    throw new Error(
-      "PythonInferenceEngine: use runJpeg() directly \u2014 this engine operates on JPEG input"
-    );
-  }
-  /** IInferenceEngine.runMultiOutput — not supported by Python engine (operates on JPEG input) */
-  async runMultiOutput(_input, _inputShape) {
-    throw new Error(
-      "PythonInferenceEngine: runMultiOutput() is not supported \u2014 this engine operates on JPEG input"
-    );
-  }
-  async dispose() {
-    if (this.process) {
-      this.process.stdin?.end();
-      this.process.kill("SIGTERM");
-      this.process = null;
-    }
-  }
-};
-
-// src/shared/engine-resolver.ts
-import * as fs from "fs";
-import * as path2 from "path";
-var AUTO_BACKEND_PRIORITY = ["coreml", "cuda", "tensorrt", "cpu"];
-var BACKEND_TO_FORMAT = {
-  cpu: "onnx",
-  coreml: "onnx",
-  cuda: "onnx",
-  tensorrt: "onnx"
-};
-var RUNTIME_TO_FORMAT = {
-  onnx: "onnx",
-  coreml: "coreml",
-  openvino: "openvino",
-  tflite: "tflite",
-  pytorch: "pt"
-};
-function modelFilePath(modelsDir, modelEntry, format) {
-  const formatEntry = modelEntry.formats[format];
-  if (!formatEntry) {
-    throw new Error(`Model ${modelEntry.id} has no ${format} format`);
-  }
-  const urlParts = formatEntry.url.split("/");
-  const filename = urlParts[urlParts.length - 1] ?? `${modelEntry.id}.${format}`;
-  return path2.join(modelsDir, filename);
-}
-function modelExists(filePath) {
-  try {
-    return fs.existsSync(filePath);
-  } catch {
-    return false;
-  }
-}
-async function resolveEngine(options) {
-  const { runtime, backend, modelEntry, modelsDir, models } = options;
-  let selectedFormat;
-  let selectedBackend;
-  if (runtime === "auto") {
-    const available = await probeOnnxBackends();
-    let chosen = null;
-    for (const b of AUTO_BACKEND_PRIORITY) {
-      if (!available.includes(b)) continue;
-      const fmt = BACKEND_TO_FORMAT[b];
-      if (!fmt) continue;
-      if (!modelEntry.formats[fmt]) continue;
-      chosen = { backend: b, format: fmt };
-      break;
-    }
-    if (!chosen) {
-      throw new Error(
-        `resolveEngine: no compatible backend found for model ${modelEntry.id}. Available backends: ${available.join(", ")}`
-      );
-    }
-    selectedFormat = chosen.format;
-    selectedBackend = chosen.backend;
-  } else {
-    const fmt = RUNTIME_TO_FORMAT[runtime];
-    if (!fmt) {
-      throw new Error(`resolveEngine: unsupported runtime "${runtime}"`);
-    }
-    if (!modelEntry.formats[fmt]) {
-      throw new Error(
-        `resolveEngine: model ${modelEntry.id} has no ${fmt} format for runtime ${runtime}`
-      );
-    }
-    selectedFormat = fmt;
-    selectedBackend = runtime === "onnx" ? backend || "cpu" : runtime;
-  }
-  let modelPath;
-  if (models) {
-    modelPath = await models.ensure(modelEntry.id, selectedFormat);
-  } else {
-    modelPath = modelFilePath(modelsDir, modelEntry, selectedFormat);
-    if (!modelExists(modelPath)) {
-      throw new Error(
-        `resolveEngine: model file not found at ${modelPath} and no model service provided`
-      );
-    }
-  }
-  if (selectedFormat === "onnx") {
-    const engine = new NodeInferenceEngine(modelPath, selectedBackend);
-    await engine.initialize();
-    return { engine, format: selectedFormat, modelPath };
-  }
-  const { pythonPath } = options;
-  const PYTHON_SCRIPT_MAP = {
-    coreml: "coreml_inference.py",
-    pytorch: "pytorch_inference.py",
-    openvino: "openvino_inference.py"
-  };
-  const effectiveRuntime = runtime === "auto" ? selectedBackend : runtime;
-  const scriptName = PYTHON_SCRIPT_MAP[effectiveRuntime];
-  if (scriptName && pythonPath) {
-    const candidates = [
-      path2.join(__dirname, "../../python", scriptName),
-      path2.join(__dirname, "../python", scriptName),
-      path2.join(__dirname, "../../../python", scriptName)
-    ];
-    const scriptPath = candidates.find((p) => fs.existsSync(p));
-    if (!scriptPath) {
-      throw new Error(
-        `resolveEngine: Python script "${scriptName}" not found. Searched:
-${candidates.join("\n")}`
-      );
-    }
-    const inputSize = Math.max(modelEntry.inputSize.width, modelEntry.inputSize.height);
-    const engine = new PythonInferenceEngine(pythonPath, scriptPath, effectiveRuntime, modelPath, [
-      `--input-size=${inputSize}`,
-      `--confidence=0.25`
-    ]);
-    await engine.initialize();
-    return { engine, format: selectedFormat, modelPath };
-  }
-  const fallbackPath = modelFilePath(modelsDir, modelEntry, "onnx");
-  if (modelEntry.formats["onnx"] && modelExists(fallbackPath)) {
-    const engine = new NodeInferenceEngine(fallbackPath, "cpu");
-    await engine.initialize();
-    return { engine, format: "onnx", modelPath: fallbackPath };
-  }
-  throw new Error(
-    `resolveEngine: format ${selectedFormat} is not yet supported by NodeInferenceEngine, no Python runtime is available, and no ONNX fallback exists`
-  );
-}
-async function probeOnnxBackends() {
-  const available = ["cpu"];
-  try {
-    const ort = await import("onnxruntime-node");
-    const providers = ort.env?.webgl?.disabled !== void 0 ? ort.InferenceSession?.getAvailableProviders?.() ?? [] : [];
-    for (const p of providers) {
-      const normalized = p.toLowerCase().replace("executionprovider", "");
-      if (normalized === "coreml") available.push("coreml");
-      else if (normalized === "cuda") available.push("cuda");
-      else if (normalized === "tensorrt") available.push("tensorrt");
-    }
-  } catch {
-  }
-  if (process.platform === "darwin" && !available.includes("coreml")) {
-    available.push("coreml");
-  }
-  return [...new Set(available)];
-}
-
-export {
-  NodeInferenceEngine,
-  PythonInferenceEngine,
-  resolveEngine,
-  probeOnnxBackends
-};
-//# sourceMappingURL=chunk-2IOKI4ES.mjs.map

package/dist/chunk-7DYHXUPZ.mjs

@@ -1,36 +0,0 @@
-import {
-  MLPACKAGE_FILES
-} from "./chunk-BP7H4NFS.mjs";
-
-// src/catalogs/segmentation-refiner-models.ts
-import { hfModelUrl } from "@camstack/types";
-var HF_REPO = "camstack/camstack-models";
-var hf = (path) => hfModelUrl(HF_REPO, path);
-var MASK_LABELS = [
-  { id: "mask", name: "Segmentation Mask" }
-];
-var SEGMENTATION_REFINER_MODELS = [
-  {
-    id: "u2netp",
-    name: "U2-Net Portable",
-    description: "U2-Net-P \u2014 ultra-lightweight salient object segmentation (4.7 MB), no prompt needed",
-    inputSize: { width: 320, height: 320 },
-    labels: MASK_LABELS,
-    formats: {
-      onnx: { url: hf("segmentationRefiner/u2netp/onnx/camstack-u2netp.onnx"), sizeMB: 5 },
-      coreml: {
-        url: hf("segmentationRefiner/u2netp/coreml/camstack-u2netp.mlpackage"),
-        sizeMB: 3,
-        isDirectory: true,
-        files: MLPACKAGE_FILES,
-        runtimes: ["python"]
-      }
-      // OpenVINO: not yet converted
-    }
-  }
-];
-
-export {
-  SEGMENTATION_REFINER_MODELS
-};
-//# sourceMappingURL=chunk-7DYHXUPZ.mjs.map

package/dist/chunk-BJTO5JO5.mjs

@@ -1,11 +0,0 @@
-var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
-  get: (a, b) => (typeof require !== "undefined" ? require : a)[b]
-}) : x)(function(x) {
-  if (typeof require !== "undefined") return require.apply(this, arguments);
-  throw Error('Dynamic require of "' + x + '" is not supported');
-});
-
-export {
-  __require
-};
-//# sourceMappingURL=chunk-BJTO5JO5.mjs.map