UNPKG

ppu-yolo-onnx-inference

Version:

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

18 lines (17 loc) 940 B
import type { InferenceSession, Tensor } from "onnxruntime-common"; import type { CoreCanvas, PlatformProvider } from "../core/platform.js"; /** * Node.js platform provider using `onnxruntime-node` and `ppu-ocv`. */ export declare class NodePlatformProvider implements PlatformProvider { createInferenceSession(model: ArrayBuffer): Promise<InferenceSession>; createTensor(type: Tensor.Type, data: Float32Array, dims: readonly number[]): Tensor; createCanvas(width: number, height: number): CoreCanvas; isCanvasLike(value: unknown): value is CoreCanvas; prepareCanvas(buffer: ArrayBuffer): Promise<CoreCanvas>; resizeImage(source: CoreCanvas, width: number, height: number): CoreCanvas; getContext2D(canvas: CoreCanvas): CanvasRenderingContext2D; saveDebugImage(canvas: CoreCanvas, filename: string, outputDir: string): Promise<void>; initRuntime(): Promise<void>; scheduleYield(): Promise<void>; }