@idscan/idvc2
Version:
component for the capturing documents
40 lines (39 loc) • 1.56 kB
TypeScript
/// <reference types="onnxruntime-web" />
import NeuralModel from './neuralModel';
import { Provider } from './Provider';
import type { ISize } from '../helpers/ts/common';
import type { GeneralTypeExitYolo } from './generalTypeYolo/exits';
import type { Bbox } from '../types/geometry';
declare const modelConfig: {
confidence: number;
predictConfidence: number;
classes: string[];
inputImageWidth: number;
inputImageHeight: number;
inputTensorSize: number[];
inputTensorName: string;
outputTensorSize: number[];
outputTensorName: string;
predictLength: number;
padding: number;
documentFramePadding: number;
};
export type InterpretYoloDetectionResult = {
side: GeneralTypeExitYolo;
bbox: Bbox;
perimeterInPercents: ISize;
isDocumentInsideFrame: boolean;
isHasFaceOnDocument: boolean;
};
export declare class GeneralTypeDetectionYoloModel extends NeuralModel {
private canvasSize;
private documentPredictLabelIndex;
private documentPredictConfidenceIndex;
constructor(url?: string, provider?: Provider, wasmPaths?: string);
private setSizes;
private findHighestConfidencePredict;
predictFromImage(input: HTMLCanvasElement | HTMLImageElement | ImageData, dstCanvas?: HTMLCanvasElement, modelConfigParam?: typeof modelConfig): Promise<Record<string, unknown> | InterpretYoloDetectionResult>;
private interpret;
predict(rgb: Uint8Array, modelConfigParam: typeof modelConfig): Promise<import("onnxruntime-web").InferenceSession.OnnxValueMapType>;
}
export {};