ppu-yolo-onnx-inference
Version:
Use your YOLO onnx object detection model in Typescript Bun environment easily.
18 lines (17 loc) • 940 B
TypeScript
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>;
}