Quick Start

import { useState } from "react";
import { parseModel } from "@wetron/core";
import type { ModelGraph } from "@wetron/core";
import { ModelGraphView, NodePropertyPanel } from "@wetron/react";
import type { PanelTarget } from "@wetron/react";
import "@wetron/react/styles.css";

export default function App() {
  const [graph, setGraph] = useState<ModelGraph | null>(null);
  const [selected, setSelected] = useState<PanelTarget | null>(null);

  async function handleFile(e: React.ChangeEvent<HTMLInputElement>) {
    const file = e.target.files?.[0];
    if (!file) return;
    const bytes = new Uint8Array(await file.arrayBuffer());
    setGraph(await parseModel(bytes, file.name));
  }

  return (
    <>
      <input type="file" accept=".onnx,.tflite,.keras,.pt,.pte,.pb" onChange={handleFile} />
      {graph && <ModelGraphView graph={graph} onTargetClick={setSelected} colorMode="system" />}
      <NodePropertyPanel
        target={selected}
        colorMode="system"
        opsets={graph?.opsets}
        tensorShapes={graph?.tensorShapes}
        onClose={() => setSelected(null)}
      />
    </>
  );
}

<script>
  import { parseModel } from "@wetron/core";
  import { ModelGraphView, NodePropertyPanel } from "@wetron/svelte";
  let graph = $state(null);
  let selected = $state(null);

  async function handleFile(e) {
    const file = e.target.files[0];
    const bytes = new Uint8Array(await file.arrayBuffer());
    graph = await parseModel(bytes, file.name);
  }
</script>

<input type="file" accept=".onnx,.tflite,.keras,.pt,.pte,.pb" onchange={handleFile} />
{#if graph}
  <ModelGraphView {graph} onTargetClick={(t) => selected = t} colorMode="system" />
{/if}
<NodePropertyPanel target={selected} colorMode="system" />

import { ParseError } from "@wetron/core";

try {
  const graph = await parseModel(bytes, file.name);
  if (graph.warnings?.length) {
    console.warn("Parse warnings:", graph.warnings);
  }
} catch (e) {
  if (e instanceof ParseError) {
    console.error(`[${e.format}] ${e.context}`);
  }
}

import { detectFormat } from "@wetron/core";

const format = detectFormat(bytes, file.name);
// "onnx" | "tflite" | "keras" | "torchscript" | "executorch" | "savedmodel" | "unknown"
// never throws

import { decodeFirstN, computeStats } from "@wetron/core";

const bytes = graph.weights?.get("conv1.weight");
if (bytes) {
  const preview = decodeFirstN(bytes, "float32", 4096);
  if (preview instanceof Float64Array) {
    const stats = computeStats(preview);
    // stats.min, stats.max, stats.mean, stats.std,
    // stats.histogram (12 bins), stats.heatmap (16x8)
  }
}

import { loadSavedModelWeights, attachCheckpointToGraph } from "@wetron/savedmodel";

if (graph.hasExternalWeights) {
  const loaded = await loadSavedModelWeights(indexFile, dataFile);
  const withWeights = attachCheckpointToGraph(graph, loaded);
  // withWeights.weights.get(nodeName) -> Uint8Array | undefined
}