use onnx model to process an image in javascript

index.tsx
npm install onnxjs
19 chars
2 lines

index.tsx
const modelPath = 'path/to/your/onnx/model.onnx';
const session = new onnx.InferenceSession();
await session.loadModel(modelPath);
131 chars
4 lines

index.tsx
// Load the image using the Canvas API
const canvas = document.createElement('canvas');
const ctx = canvas.getContext('2d');
const img = new Image();
img.src = 'path/to/your/image.jpg';
await new Promise(resolve => img.onload = resolve);
canvas.width = img.width;
canvas.height = img.height;
ctx.drawImage(img, 0, 0);

// Convert the image to a tensor
const imageData = ctx.getImageData(0, 0, img.width, img.height).data;
const pixels = new Float32Array(imageData).map(x => x / 255);
const input = new onnx.Tensor(pixels, 'float32', [1, 3, img.height, img.width]);
565 chars
15 lines

index.tsx
const outputMap = await session.run([input]);
const outputData = outputMap.values().next().value.data;
103 chars
3 lines