UNPKG

ppu-yolo-onnx-inference

Version:

Use your YOLO onnx object detection model in Typescript Bun environment easily.

1 lines 1.03 kB
export class NodePlatformProvider{createInferenceSession(model){return ort.InferenceSession.create(model)}createTensor(type,data,dims){return new ort.Tensor(type,data,dims)}createCanvas(width,height){return createCanvas(width,height)}isCanvasLike(value){return value!=null&&typeof value==="object"&&"width"in value&&"height"in value&&"getContext"in value&&typeof value.getContext==="function"}async prepareCanvas(buffer){return ImageProcessor.prepareCanvas(buffer)}resizeImage(source,width,height){let processor=new ImageProcessor(source);let resized=processor.resize({width,height}).toCanvas();processor.destroy();return resized}getContext2D(canvas){return canvas.getContext("2d")}async saveDebugImage(canvas,filename,outputDir){await CanvasToolkit.getInstance().saveImage({canvas,filename,path:outputDir})}async initRuntime(){await ImageProcessor.initRuntime()}scheduleYield(){return new Promise((resolve)=>setImmediate(resolve))}}import*as ort from"onnxruntime-node";import{CanvasToolkit,createCanvas,ImageProcessor}from"ppu-ocv";