@areb0s/ocr-common 1.0.0

package/build/Ocr.d.ts

@@ -0,0 +1,15 @@
+import type { ImageRawData, BrowserImageInput, ModelCreateOptions } from './types/index.js';
+import { Detection, Recognition } from './models/index.js';
+export declare class Ocr {
+    #private;
+    static create(options?: ModelCreateOptions): Promise<Ocr>;
+    constructor({ detection, recognition, }: {
+        detection: Detection;
+        recognition: Recognition;
+    });
+    detect(image: string | ImageRawData | BrowserImageInput, options?: {}): Promise<{
+        texts: import("./types/index.js").Line[];
+        resizedImageWidth: any;
+        resizedImageHeight: any;
+    }>;
+}

package/build/Ocr.js

@@ -0,0 +1,24 @@
+import { Detection, Recognition } from './models/index.js';
+export class Ocr {
+    static async create(options = {}) {
+        const detection = await Detection.create(options);
+        const recognition = await Recognition.create(options);
+        return new Ocr({ detection, recognition });
+    }
+    #detection;
+    #recognition;
+    constructor({ detection, recognition, }) {
+        this.#detection = detection;
+        this.#recognition = recognition;
+    }
+    async detect(image, options = {}) {
+        const { lineImages, resizedImageWidth, resizedImageHeight } = await this.#detection.run(image, options);
+        const texts = await this.#recognition.run(lineImages, options);
+        return {
+            texts,
+            resizedImageWidth,
+            resizedImageHeight,
+        };
+    }
+}
+//# sourceMappingURL=Ocr.js.map

package/build/Ocr.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Ocr.js","sourceRoot":"","sources":["../src/Ocr.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,UAAU,CAAA;AAEjD,MAAM,OAAO,GAAG;IACd,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,UAA8B,EAAE;QAClD,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QACjD,MAAM,WAAW,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QACrD,OAAO,IAAI,GAAG,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAA;IAC5C,CAAC;IAED,UAAU,CAAW;IACrB,YAAY,CAAa;IAEzB,YAAY,EACV,SAAS,EACT,WAAW,GAIZ;QACC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAA;QAC3B,IAAI,CAAC,YAAY,GAAG,WAAW,CAAA;IACjC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAgD,EAAE,OAAO,GAAG,EAAE;QACzE,MAAM,EAAE,UAAU,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAA;QACvG,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC,CAAA;QAC9D,OAAO;YACL,KAAK;YACL,iBAAiB;YACjB,kBAAkB;SACnB,CAAA;IACH,CAAC;CACF"}

package/build/backend/FileUtilsBase.d.ts

@@ -0,0 +1,3 @@
+export declare class FileUtilsBase {
+    static read(path: string): Promise<string>;
+}

package/build/backend/FileUtilsBase.js

@@ -0,0 +1,6 @@
+export class FileUtilsBase {
+    static async read(path) {
+        throw new Error('Not implemented');
+    }
+}
+//# sourceMappingURL=FileUtilsBase.js.map

package/build/backend/FileUtilsBase.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FileUtilsBase.js","sourceRoot":"","sources":["../../src/backend/FileUtilsBase.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,aAAa;IACxB,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,IAAY;QAC5B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAA;IACpC,CAAC;CACF"}

package/build/backend/ImageRawBase.d.ts

@@ -0,0 +1,8 @@
+import type { ImageRawData } from '../types/index.js';
+export declare class ImageRawBase {
+    data: ImageRawData['data'];
+    width: ImageRawData['width'];
+    height: ImageRawData['height'];
+    constructor({ data, width, height }: ImageRawData);
+    getImageRawData(): ImageRawData;
+}

package/build/backend/ImageRawBase.js

@@ -0,0 +1,14 @@
+export class ImageRawBase {
+    data;
+    width;
+    height;
+    constructor({ data, width, height }) {
+        this.data = data;
+        this.width = width;
+        this.height = height;
+    }
+    getImageRawData() {
+        return { data: this.data, width: this.width, height: this.height };
+    }
+}
+//# sourceMappingURL=ImageRawBase.js.map

package/build/backend/ImageRawBase.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ImageRawBase.js","sourceRoot":"","sources":["../../src/backend/ImageRawBase.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,YAAY;IACvB,IAAI,CAAsB;IAC1B,KAAK,CAAuB;IAC5B,MAAM,CAAwB;IAE9B,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAgB;QAC/C,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAA;QAClB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACtB,CAAC;IAED,eAAe;QACb,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAA;IACpE,CAAC;CACF"}

package/build/backend/backend.d.ts

@@ -0,0 +1,14 @@
+import type { FileUtils as FileUtilsType, ImageRaw as ImageRawType, InferenceSession as InferenceSessionType, ModelCreateOptions as ModelCreateOptionsType, SplitIntoLineImages as SplitIntoLineImagesType } from '../types/index.js';
+declare let FileUtils: FileUtilsType | any;
+declare let ImageRaw: ImageRawType | any;
+declare let InferenceSession: InferenceSessionType | any;
+declare let splitIntoLineImages: SplitIntoLineImagesType;
+declare let defaultModels: ModelCreateOptionsType['models'];
+export declare function registerBackend(backend: {
+    FileUtils: FileUtilsType;
+    ImageRaw: ImageRawType | any;
+    InferenceSession: InferenceSessionType | any;
+    splitIntoLineImages: SplitIntoLineImagesType;
+    defaultModels: ModelCreateOptionsType['models'];
+}): void;
+export { FileUtils, ImageRaw, InferenceSession, splitIntoLineImages, defaultModels };

package/build/backend/backend.js

@@ -0,0 +1,14 @@
+let FileUtils = undefined;
+let ImageRaw = undefined;
+let InferenceSession = undefined;
+let splitIntoLineImages = undefined;
+let defaultModels = undefined;
+export function registerBackend(backend) {
+    FileUtils = backend.FileUtils;
+    ImageRaw = backend.ImageRaw;
+    InferenceSession = backend.InferenceSession;
+    splitIntoLineImages = backend.splitIntoLineImages;
+    defaultModels = backend.defaultModels;
+}
+export { FileUtils, ImageRaw, InferenceSession, splitIntoLineImages, defaultModels };
+//# sourceMappingURL=backend.js.map

package/build/backend/backend.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"backend.js","sourceRoot":"","sources":["../../src/backend/backend.ts"],"names":[],"mappings":"AAQA,IAAI,SAAS,GAAwB,SAAqC,CAAA;AAC1E,IAAI,QAAQ,GAAuB,SAAoC,CAAA;AACvE,IAAI,gBAAgB,GAA+B,SAA4C,CAAA;AAC/F,IAAI,mBAAmB,GAA4B,SAA+C,CAAA;AAClG,IAAI,aAAa,GAAqC,SAAwD,CAAA;AAE9G,MAAM,UAAU,eAAe,CAAC,OAM/B;IACC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAA;IAC7B,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAA;IAC3B,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAA;IAC3C,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,CAAA;IACjD,aAAa,GAAG,OAAO,CAAC,aAAa,CAAA;AACvC,CAAC;AAED,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,aAAa,EAAE,CAAA"}

package/build/backend/index.d.ts

@@ -0,0 +1,3 @@
+export * from './backend.js';
+export * from './FileUtilsBase.js';
+export * from './ImageRawBase.js';

package/build/backend/index.js

@@ -0,0 +1,4 @@
+export * from './backend.js';
+export * from './FileUtilsBase.js';
+export * from './ImageRawBase.js';
+//# sourceMappingURL=index.js.map

package/build/backend/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/backend/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAA;AACzB,cAAc,iBAAiB,CAAA;AAC/B,cAAc,gBAAgB,CAAA"}

package/build/backend/splitIntoLineImages.d.ts

@@ -0,0 +1,2 @@
+import type { LineImage, ImageRaw as ImageRawType } from '../types/index.js';
+export declare function splitIntoLineImages(image: ImageRawType, sourceImage: ImageRawType): Promise<LineImage[]>;

package/build/backend/splitIntoLineImages.js

@@ -0,0 +1,231 @@
+import cv from '@techstark/opencv-js';
+import clipper from 'js-clipper';
+import { ImageRaw } from '../backend/index.js';
+export async function splitIntoLineImages(image, sourceImage) {
+    const w = image.width;
+    const h = image.height;
+    const srcData = sourceImage;
+    const edgeRect = [];
+    const src = cvImread(image);
+    cv.cvtColor(src, src, cv.COLOR_RGBA2GRAY, 0);
+    const contours = new cv.MatVector();
+    const hierarchy = new cv.Mat();
+    cv.findContours(src, contours, hierarchy, cv.RETR_LIST, cv.CHAIN_APPROX_SIMPLE);
+    for (let i = 0; i < contours.size(); i++) {
+        const minSize = 3;
+        const cnt = contours.get(i);
+        const { points, sside } = getMiniBoxes(cnt);
+        if (sside < minSize)
+            continue;
+        // TODO sort fast
+        const clipBox = unclip(points);
+        const boxMap = cv.matFromArray(clipBox.length / 2, 1, cv.CV_32SC2, clipBox);
+        const resultObj = getMiniBoxes(boxMap);
+        const box = resultObj.points;
+        if (resultObj.sside < minSize + 2) {
+            continue;
+        }
+        function clip(n, min, max) {
+            return Math.max(min, Math.min(n, max));
+        }
+        const rx = srcData.width / w;
+        const ry = srcData.height / h;
+        for (let i = 0; i < box.length; i++) {
+            box[i][0] *= rx;
+            box[i][1] *= ry;
+        }
+        const box1 = orderPointsClockwise(box);
+        box1.forEach((item) => {
+            item[0] = clip(Math.round(item[0]), 0, srcData.width);
+            item[1] = clip(Math.round(item[1]), 0, srcData.height);
+        });
+        const rect_width = int(linalgNorm(box1[0], box1[1]));
+        const rect_height = int(linalgNorm(box1[0], box1[3]));
+        if (rect_width <= 3 || rect_height <= 3)
+            continue;
+        const c = getRotateCropImage(srcData, box);
+        edgeRect.push({
+            box,
+            image: c,
+        });
+    }
+    src.delete();
+    contours.delete();
+    hierarchy.delete();
+    return edgeRect;
+}
+function getMiniBoxes(contour) {
+    const boundingBox = cv.minAreaRect(contour);
+    const points = Array.from(boxPoints(boundingBox.center, boundingBox.size, boundingBox.angle)).sort((a, b) => a[0] - b[0]);
+    let index_1 = 0, index_2 = 1, index_3 = 2, index_4 = 3;
+    if (points[1][1] > points[0][1]) {
+        index_1 = 0;
+        index_4 = 1;
+    }
+    else {
+        index_1 = 1;
+        index_4 = 0;
+    }
+    if (points[3][1] > points[2][1]) {
+        index_2 = 2;
+        index_3 = 3;
+    }
+    else {
+        index_2 = 3;
+        index_3 = 2;
+    }
+    const box = [points[index_1], points[index_2], points[index_3], points[index_4]];
+    const side = Math.min(boundingBox.size.height, boundingBox.size.width);
+    return { points: box, sside: side };
+}
+function unclip(box) {
+    const unclip_ratio = 1.5;
+    const area = Math.abs(polygonPolygonArea(box));
+    const length = polygonPolygonLength(box);
+    const distance = (area * unclip_ratio) / length;
+    const tmpArr = [];
+    box.forEach((item) => {
+        const obj = {
+            X: 0,
+            Y: 0,
+        };
+        obj.X = item[0];
+        obj.Y = item[1];
+        tmpArr.push(obj);
+    });
+    const offset = new clipper.ClipperOffset();
+    offset.AddPath(tmpArr, clipper.JoinType.jtRound, clipper.EndType.etClosedPolygon);
+    const expanded = [];
+    offset.Execute(expanded, distance);
+    let expandedArr = [];
+    expanded[0] &&
+        expanded[0].forEach((item) => {
+            expandedArr.push([item.X, item.Y]);
+        });
+    expandedArr = [].concat(...expandedArr);
+    return expandedArr;
+}
+function orderPointsClockwise(pts) {
+    const rect = [
+        [0, 0],
+        [0, 0],
+        [0, 0],
+        [0, 0],
+    ];
+    const s = pts.map((pt) => pt[0] + pt[1]);
+    rect[0] = pts[s.indexOf(Math.min(...s))];
+    rect[2] = pts[s.indexOf(Math.max(...s))];
+    const tmp = pts.filter((pt) => pt !== rect[0] && pt !== rect[2]);
+    const diff = tmp[1].map((e, i) => e - tmp[0][i]);
+    rect[1] = tmp[diff.indexOf(Math.min(...diff))];
+    rect[3] = tmp[diff.indexOf(Math.max(...diff))];
+    return rect;
+}
+function linalgNorm(p0, p1) {
+    return Math.sqrt(Math.pow(p0[0] - p1[0], 2) + Math.pow(p0[1] - p1[1], 2));
+}
+function int(num) {
+    return num > 0 ? Math.floor(num) : Math.ceil(num);
+}
+function getRotateCropImage(imageRaw, points) {
+    const img_crop_width = int(Math.max(linalgNorm(points[0], points[1]), linalgNorm(points[2], points[3])));
+    const img_crop_height = int(Math.max(linalgNorm(points[0], points[3]), linalgNorm(points[1], points[2])));
+    const pts_std = [
+        [0, 0],
+        [img_crop_width, 0],
+        [img_crop_width, img_crop_height],
+        [0, img_crop_height],
+    ];
+    const srcTri = cv.matFromArray(4, 1, cv.CV_32FC2, flatten(points));
+    const dstTri = cv.matFromArray(4, 1, cv.CV_32FC2, flatten(pts_std));
+    // 获取到目标矩阵
+    const M = cv.getPerspectiveTransform(srcTri, dstTri);
+    const src = cvImread(imageRaw);
+    const dst = new cv.Mat();
+    const dsize = new cv.Size(img_crop_width, img_crop_height);
+    // 透视转换
+    cv.warpPerspective(src, dst, M, dsize, cv.INTER_CUBIC, cv.BORDER_REPLICATE, new cv.Scalar());
+    const dst_img_height = dst.matSize[0];
+    const dst_img_width = dst.matSize[1];
+    let dst_rot;
+    // 图像旋转
+    if (dst_img_height / dst_img_width >= 1.5) {
+        dst_rot = new cv.Mat();
+        const dsize_rot = new cv.Size(dst.rows, dst.cols);
+        const center = new cv.Point(dst.cols / 2, dst.cols / 2);
+        const M = cv.getRotationMatrix2D(center, 90, 1);
+        cv.warpAffine(dst, dst_rot, M, dsize_rot, cv.INTER_CUBIC, cv.BORDER_REPLICATE, new cv.Scalar());
+    }
+    src.delete();
+    srcTri.delete();
+    dstTri.delete();
+    if (dst_rot) {
+        dst.delete();
+    }
+    return cvImshow(dst_rot || dst);
+}
+function boxPoints(center, size, angle) {
+    const width = size.width;
+    const height = size.height;
+    const theta = (angle * Math.PI) / 180.0;
+    const cosTheta = Math.cos(theta);
+    const sinTheta = Math.sin(theta);
+    const cx = center.x;
+    const cy = center.y;
+    const dx = width * 0.5;
+    const dy = height * 0.5;
+    const rotatedPoints = [];
+    // Top-Left
+    const x1 = cx - dx * cosTheta + dy * sinTheta;
+    const y1 = cy - dx * sinTheta - dy * cosTheta;
+    rotatedPoints.push([x1, y1]);
+    // Top-Right
+    const x2 = cx + dx * cosTheta + dy * sinTheta;
+    const y2 = cy + dx * sinTheta - dy * cosTheta;
+    rotatedPoints.push([x2, y2]);
+    // Bottom-Right
+    const x3 = cx + dx * cosTheta - dy * sinTheta;
+    const y3 = cy + dx * sinTheta + dy * cosTheta;
+    rotatedPoints.push([x3, y3]);
+    // Bottom-Left
+    const x4 = cx - dx * cosTheta - dy * sinTheta;
+    const y4 = cy - dx * sinTheta + dy * cosTheta;
+    rotatedPoints.push([x4, y4]);
+    return rotatedPoints;
+}
+function polygonPolygonArea(polygon) {
+    let i = -1, n = polygon.length, a, b = polygon[n - 1], area = 0;
+    while (++i < n) {
+        a = b;
+        b = polygon[i];
+        area += a[1] * b[0] - a[0] * b[1];
+    }
+    return area / 2;
+}
+function polygonPolygonLength(polygon) {
+    let i = -1, n = polygon.length, b = polygon[n - 1], xa, ya, xb = b[0], yb = b[1], perimeter = 0;
+    while (++i < n) {
+        xa = xb;
+        ya = yb;
+        b = polygon[i];
+        xb = b[0];
+        yb = b[1];
+        xa -= xb;
+        ya -= yb;
+        perimeter += Math.hypot(xa, ya);
+    }
+    return perimeter;
+}
+function flatten(arr) {
+    return arr
+        .toString()
+        .split(',')
+        .map((item) => +item);
+}
+function cvImread(image) {
+    return cv.matFromImageData(image);
+}
+function cvImshow(mat) {
+    return new ImageRaw({ data: mat.data, width: mat.cols, height: mat.rows });
+}
+//# sourceMappingURL=splitIntoLineImages.js.map

package/build/backend/splitIntoLineImages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"splitIntoLineImages.js","sourceRoot":"","sources":["../../src/backend/splitIntoLineImages.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,sBAAsB,CAAA;AACrC,OAAO,OAAO,MAAM,YAAY,CAAA;AAChC,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAA;AAO1C,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,KAAmB,EAAE,WAAyB;IACtF,MAAM,CAAC,GAAG,KAAK,CAAC,KAAK,CAAA;IACrB,MAAM,CAAC,GAAG,KAAK,CAAC,MAAM,CAAA;IACtB,MAAM,OAAO,GAAG,WAAW,CAAA;IAE3B,MAAM,QAAQ,GAA4C,EAAE,CAAA;IAE5D,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAA;IAE3B,EAAE,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,eAAe,EAAE,CAAC,CAAC,CAAA;IAC5C,MAAM,QAAQ,GAAG,IAAI,EAAE,CAAC,SAAS,EAAE,CAAA;IACnC,MAAM,SAAS,GAAG,IAAI,EAAE,CAAC,GAAG,EAAE,CAAA;IAE9B,EAAE,CAAC,YAAY,CAAC,GAAG,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC,mBAAmB,CAAC,CAAA;IAE/E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,CAAC,CAAA;QACjB,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAA;QAC3B,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,YAAY,CAAC,GAAG,CAAC,CAAA;QAC3C,IAAI,KAAK,GAAG,OAAO;YAAE,SAAQ;QAC7B,iBAAiB;QAEjB,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAA;QAE9B,MAAM,MAAM,GAAG,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;QAE3E,MAAM,SAAS,GAAG,YAAY,CAAC,MAAM,CAAC,CAAA;QACtC,MAAM,GAAG,GAAG,SAAS,CAAC,MAAM,CAAA;QAC5B,IAAI,SAAS,CAAC,KAAK,GAAG,OAAO,GAAG,CAAC,EAAE,CAAC;YAClC,SAAQ;QACV,CAAC;QACD,SAAS,IAAI,CAAC,CAAS,EAAE,GAAW,EAAE,GAAW;YAC/C,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAA;QACxC,CAAC;QAED,MAAM,EAAE,GAAG,OAAO,CAAC,KAAK,GAAG,CAAC,CAAA;QAC5B,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAA;QAE7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;YACf,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;QACjB,CAAC;QAED,MAAM,IAAI,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAA;QACtC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YACpB,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,KAAK,CAAC,CAAA;YACrD,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;QACxD,CAAC,CAAC,CAAA;QACF,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;QACpD,MAAM,WAAW,GAAG,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;QACrD,IAAI,UAAU,IAAI,CAAC,IAAI,WAAW,IAAI,CAAC;YAAE,SAAQ;QAEjD,MAAM,CAAC,GAAG,kBAAkB,CAAC,OAAO,EAAE,GAAG,CAAC,CAAA;QAE1C,QAAQ,CAAC,IAAI,CAAC;YACZ,GAAG;YACH,KAAK,EAAE,CAAC;SACT,CAAC,CAAA;IACJ,CAAC;IAED,GAAG,CAAC,MAAM,EAAE,CAAA;IACZ,QAAQ,CAAC,MAAM,EAAE,CAAA;IACjB,SAAS,CAAC,MAAM,EAAE,CAAA;IAElB,OAAO,QAAQ,CAAA;AACjB,CAAC;AAED,SAAS,YAAY,CAAC,OAAY;IAChC,MAAM,WAAW,GAAG,EAAE,CAAC,WAAW,CAAC,OAAO,CAAC,CAAA;IAC3C,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,EAAE,WAAW,CAAC,IAAI,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAChG,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CACR,CAAA;IAEf,IAAI,OAAO,GAAG,CAAC,EACb,OAAO,GAAG,CAAC,EACX,OAAO,GAAG,CAAC,EACX,OAAO,GAAG,CAAC,CAAA;IACb,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAChC,OAAO,GAAG,CAAC,CAAA;QACX,OAAO,GAAG,CAAC,CAAA;IACb,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,CAAC,CAAA;QACX,OAAO,GAAG,CAAC,CAAA;IACb,CAAC;IACD,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAChC,OAAO,GAAG,CAAC,CAAA;QACX,OAAO,GAAG,CAAC,CAAA;IACb,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,CAAC,CAAA;QACX,OAAO,GAAG,CAAC,CAAA;IACb,CAAC;IAED,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAY,CAAA;IAC3F,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACtE,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,CAAA;AACrC,CAAC;AAED,SAAS,MAAM,CAAC,GAAe;IAC7B,MAAM,YAAY,GAAG,GAAG,CAAA;IACxB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,CAAA;IAC9C,MAAM,MAAM,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAA;IACxC,MAAM,QAAQ,GAAG,CAAC,IAAI,GAAG,YAAY,CAAC,GAAG,MAAM,CAAA;IAC/C,MAAM,MAAM,GAA+B,EAAE,CAAA;IAC7C,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QACnB,MAAM,GAAG,GAAG;YACV,CAAC,EAAE,CAAC;YACJ,CAAC,EAAE,CAAC;SACL,CAAA;QACD,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;QACf,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;QACf,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IAClB,CAAC,CAAC,CAAA;IACF,MAAM,MAAM,GAAG,IAAI,OAAO,CAAC,aAAa,EAAE,CAAA;IAC1C,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,CAAA;IACjF,MAAM,QAAQ,GAAiC,EAAE,CAAA;IACjD,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAA;IAClC,IAAI,WAAW,GAAe,EAAE,CAAA;IAChC,QAAQ,CAAC,CAAC,CAAC;QACT,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YAC3B,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;QACpC,CAAC,CAAC,CAAA;IACJ,WAAW,GAAG,EAAE,CAAC,MAAM,CAAC,GAAS,WAAY,CAAC,CAAA;IAE9C,OAAO,WAAW,CAAA;AACpB,CAAC;AAED,SAAS,oBAAoB,CAAC,GAAY;IACxC,MAAM,IAAI,GAAY;QACpB,CAAC,CAAC,EAAE,CAAC,CAAC;QACN,CAAC,CAAC,EAAE,CAAC,CAAC;QACN,CAAC,CAAC,EAAE,CAAC,CAAC;QACN,CAAC,CAAC,EAAE,CAAC,CAAC;KACP,CAAA;IACD,MAAM,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;IACxC,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;IACxC,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;IACxC,MAAM,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;IAChE,MAAM,IAAI,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IAChD,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;IAC9C,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;IAC9C,OAAO,IAAI,CAAA;AACb,CAAC;AAED,SAAS,UAAU,CAAC,EAAa,EAAE,EAAa;IAC9C,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;AAC3E,CAAC;AAED,SAAS,GAAG,CAAC,GAAW;IACtB,OAAO,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AACnD,CAAC;AAED,SAAS,kBAAkB,CAAC,QAAsB,EAAE,MAAe;IACjE,MAAM,cAAc,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IACxG,MAAM,eAAe,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IACzG,MAAM,OAAO,GAAG;QACd,CAAC,CAAC,EAAE,CAAC,CAAC;QACN,CAAC,cAAc,EAAE,CAAC,CAAC;QACnB,CAAC,cAAc,EAAE,eAAe,CAAC;QACjC,CAAC,CAAC,EAAE,eAAe,CAAC;KACrB,CAAA;IAED,MAAM,MAAM,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAA;IAClE,MAAM,MAAM,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,CAAA;IAEnE,UAAU;IACV,MAAM,CAAC,GAAG,EAAE,CAAC,uBAAuB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IACpD,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAA;IAC9B,MAAM,GAAG,GAAG,IAAI,EAAE,CAAC,GAAG,EAAE,CAAA;IACxB,MAAM,KAAK,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,cAAc,EAAE,eAAe,CAAC,CAAA;IAC1D,OAAO;IACP,EAAE,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,gBAAgB,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,CAAC,CAAA;IAE5F,MAAM,cAAc,GAAS,GAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;IAC5C,MAAM,aAAa,GAAS,GAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;IAC3C,IAAI,OAAO,CAAA;IACX,OAAO;IACP,IAAI,cAAc,GAAG,aAAa,IAAI,GAAG,EAAE,CAAC;QAC1C,OAAO,GAAG,IAAI,EAAE,CAAC,GAAG,EAAE,CAAA;QACtB,MAAM,SAAS,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,CAAA;QACjD,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,EAAE,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,CAAA;QACvD,MAAM,CAAC,GAAG,EAAE,CAAC,mBAAmB,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;QAC/C,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,gBAAgB,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,CAAC,CAAA;IACjG,CAAC;IAED,GAAG,CAAC,MAAM,EAAE,CAAA;IACZ,MAAM,CAAC,MAAM,EAAE,CAAA;IACf,MAAM,CAAC,MAAM,EAAE,CAAA;IACf,IAAI,OAAO,EAAE,CAAC;QACZ,GAAG,CAAC,MAAM,EAAE,CAAA;IACd,CAAC;IAED,OAAO,QAAQ,CAAC,OAAO,IAAI,GAAG,CAAC,CAAA;AACjC,CAAC;AAED,SAAS,SAAS,CAAC,MAAgC,EAAE,IAAuC,EAAE,KAAa;IACzG,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;IACxB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;IAE1B,MAAM,KAAK,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,KAAK,CAAA;IACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;IAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;IAEhC,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAA;IACnB,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAA;IAEnB,MAAM,EAAE,GAAG,KAAK,GAAG,GAAG,CAAA;IACtB,MAAM,EAAE,GAAG,MAAM,GAAG,GAAG,CAAA;IAEvB,MAAM,aAAa,GAAU,EAAE,CAAA;IAE/B,WAAW;IACX,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,QAAQ,GAAG,EAAE,GAAG,QAAQ,CAAA;IAC7C,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,QAAQ,GAAG,EAAE,GAAG,QAAQ,CAAA;IAC7C,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;IAE5B,YAAY;IACZ,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,QAAQ,GAAG,EAAE,GAAG,QAAQ,CAAA;IAC7C,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,QAAQ,GAAG,EAAE,GAAG,QAAQ,CAAA;IAC7C,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;IAE5B,eAAe;IACf,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,QAAQ,GAAG,EAAE,GAAG,QAAQ,CAAA;IAC7C,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,QAAQ,GAAG,EAAE,GAAG,QAAQ,CAAA;IAC7C,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;IAE5B,cAAc;IACd,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,QAAQ,GAAG,EAAE,GAAG,QAAQ,CAAA;IAC7C,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,QAAQ,GAAG,EAAE,GAAG,QAAQ,CAAA;IAC7C,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;IAE5B,OAAO,aAAa,CAAA;AACtB,CAAC;AAED,SAAS,kBAAkB,CAAC,OAAmB;IAC7C,IAAI,CAAC,GAAG,CAAC,CAAC,EACR,CAAC,GAAG,OAAO,CAAC,MAAM,EAClB,CAAY,EACZ,CAAC,GAAG,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,EAClB,IAAI,GAAG,CAAC,CAAA;IAEV,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC;QACf,CAAC,GAAG,CAAC,CAAA;QACL,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAA;QACd,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;IACnC,CAAC;IAED,OAAO,IAAI,GAAG,CAAC,CAAA;AACjB,CAAC;AAED,SAAS,oBAAoB,CAAC,OAAmB;IAC/C,IAAI,CAAC,GAAG,CAAC,CAAC,EACR,CAAC,GAAG,OAAO,CAAC,MAAM,EAClB,CAAC,GAAG,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,EAClB,EAAU,EACV,EAAU,EACV,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,EACT,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,EACT,SAAS,GAAG,CAAC,CAAA;IAEf,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC;QACf,EAAE,GAAG,EAAE,CAAA;QACP,EAAE,GAAG,EAAE,CAAA;QACP,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAA;QACd,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;QACT,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;QACT,EAAE,IAAI,EAAE,CAAA;QACR,EAAE,IAAI,EAAE,CAAA;QACR,SAAS,IAAI,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAA;IACjC,CAAC;IAED,OAAO,SAAS,CAAA;AAClB,CAAC;AAED,SAAS,OAAO,CAAC,GAA0B;IACzC,OAAO,GAAG;SACP,QAAQ,EAAE;SACV,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAA;AACzB,CAAC;AAED,SAAS,QAAQ,CAAC,KAAmB;IACnC,OAAO,EAAE,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAA;AACnC,CAAC;AAED,SAAS,QAAQ,CAAC,GAAW;IAC3B,OAAO,IAAI,QAAQ,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC,CAAA;AAC5E,CAAC"}

package/build/index.d.ts

@@ -0,0 +1,6 @@
+import { Ocr } from './Ocr.js';
+export default Ocr;
+export { registerBackend } from './backend/backend.js';
+export * from './backend/FileUtilsBase.js';
+export * from './backend/ImageRawBase.js';
+export type * from './types/index.js';

package/build/index.js

@@ -0,0 +1,6 @@
+import { Ocr } from './Ocr.js';
+export default Ocr;
+export { registerBackend } from './backend/backend.js';
+export * from './backend/FileUtilsBase.js';
+export * from './backend/ImageRawBase.js';
+//# sourceMappingURL=index.js.map

package/build/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAA;AAE3B,eAAe,GAAG,CAAA;AAElB,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAA;AACnD,cAAc,yBAAyB,CAAA;AACvC,cAAc,wBAAwB,CAAA"}

package/build/models/Detection.d.ts

@@ -0,0 +1,13 @@
+import type { InferenceSession as InferenceSessionCommon } from 'onnxruntime-common';
+import type { ImageRawData, BrowserImageInput, ModelCreateOptions } from '../types/index.js';
+import { ModelBase } from './ModelBase.js';
+export declare class Detection extends ModelBase {
+    static create({ models, onnxOptions, ...restOptions }: ModelCreateOptions): Promise<Detection>;
+    run(input: string | ImageRawData | BrowserImageInput, { onnxOptions }?: {
+        onnxOptions?: InferenceSessionCommon.RunOptions;
+    }): Promise<{
+        lineImages: import("../types/index.js").LineImage[];
+        resizedImageWidth: any;
+        resizedImageHeight: any;
+    }>;
+}

package/build/models/Detection.js

@@ -0,0 +1,82 @@
+import invariant from 'tiny-invariant';
+import { ImageRaw, InferenceSession, defaultModels, splitIntoLineImages } from '../backend/index.js';
+import { ModelBase } from './ModelBase.js';
+const BASE_SIZE = 32;
+export class Detection extends ModelBase {
+    static async create({ models, onnxOptions = {}, ...restOptions }) {
+        const detectionPath = models?.detectionPath || defaultModels?.detectionPath;
+        invariant(detectionPath, 'detectionPath is required');
+        const model = await InferenceSession.create(detectionPath, onnxOptions);
+        return new Detection({ model, options: restOptions });
+    }
+    async run(input, { onnxOptions = {} } = {}) {
+        // Use ImageRaw.from() factory method if available (browser), otherwise fallback to legacy handling
+        const image = typeof ImageRaw.from === 'function'
+            ? await ImageRaw.from(input)
+            : typeof input === "string"
+                ? await ImageRaw.open(input)
+                : new ImageRaw(input);
+        // Resize image to multiple of 32
+        //   - image width and height must be a multiple of 32
+        //   - bigger image -> more accurate result, but takes longer time
+        // inputImage = await Image.resize(image, multipleOfBaseSize(image, { maxSize: 960 }))
+        const inputImage = await image.resize(multipleOfBaseSize(image));
+        this.debugImage(inputImage, 'out1-multiple-of-base-size.jpg');
+        // Covert image data to model data
+        //   - Using `(RGB / 255 - mean) / std` formula
+        //   - omit reshapeOptions (mean/std) is more accurate, can creaet a run option for them
+        const modelData = this.imageToInput(inputImage, {
+        // mean: [0.485, 0.456, 0.406],
+        // std: [0.229, 0.224, 0.225],
+        });
+        // Run the model
+        // console.time('Detection')
+        const modelOutput = await this.runModel({ modelData, onnxOptions });
+        // console.timeEnd('Detection')
+        // Convert output data back to image data
+        //   - output value is from 0 to 1, a probability, if value > 0.3, it is a text
+        //   - returns a black and white image
+        const outputImage = outputToImage(modelOutput, 0.03);
+        this.debugImage(outputImage, 'out2-black-white.jpg');
+        // Find text boxes, split image into lines
+        //   - findContours from the image
+        //   - returns text boxes and line images
+        const lineImages = await splitIntoLineImages(outputImage, inputImage);
+        this.debugBoxImage(inputImage, lineImages, 'boxes.jpg');
+        return {
+            lineImages,
+            resizedImageWidth: inputImage.width,
+            resizedImageHeight: inputImage.height,
+        };
+    }
+}
+function multipleOfBaseSize(image, { maxSize } = {}) {
+    let width = image.width;
+    let height = image.height;
+    if (maxSize && Math.max(width, height) > maxSize) {
+        const ratio = width > height ? maxSize / width : maxSize / height;
+        width = width * ratio;
+        height = height * ratio;
+    }
+    const newWidth = Math.max(
+    // Math.round
+    // Math.ceil
+    Math.ceil(width / BASE_SIZE) * BASE_SIZE, BASE_SIZE);
+    const newHeight = Math.max(Math.ceil(height / BASE_SIZE) * BASE_SIZE, BASE_SIZE);
+    return { width: newWidth, height: newHeight };
+}
+function outputToImage(output, threshold) {
+    const height = output.dims[2];
+    const width = output.dims[3];
+    const data = new Uint8Array(width * height * 4);
+    for (const [outIndex, outValue] of output.data.entries()) {
+        const n = outIndex * 4;
+        const value = outValue > threshold ? 255 : 0;
+        data[n] = value; // R
+        data[n + 1] = value; // G
+        data[n + 2] = value; // B
+        data[n + 3] = 255; // A
+    }
+    return new ImageRaw({ data, width, height });
+}
+//# sourceMappingURL=Detection.js.map

package/build/models/Detection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Detection.js","sourceRoot":"","sources":["../../src/models/Detection.ts"],"names":[],"mappings":"AACA,OAAO,SAAS,MAAM,gBAAgB,CAAA;AACtC,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAA;AAEhG,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAA;AAEvC,MAAM,SAAS,GAAG,EAAE,CAAA;AAEpB,MAAM,OAAO,SAAU,SAAQ,SAAS;IACtC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,WAAW,GAAG,EAAE,EAAE,GAAG,WAAW,EAAsB;QAClF,MAAM,aAAa,GAAG,MAAM,EAAE,aAAa,IAAI,aAAa,EAAE,aAAa,CAAA;QAC3E,SAAS,CAAC,aAAa,EAAE,2BAA2B,CAAC,CAAA;QACrD,MAAM,KAAK,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,aAAa,EAAE,WAAW,CAAC,CAAA;QACvE,OAAO,IAAI,SAAS,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAA;IACvD,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,KAAgD,EAAE,EAAE,WAAW,GAAG,EAAE,KAA0D,EAAE;QACxI,mGAAmG;QACnG,MAAM,KAAK,GAAG,OAAQ,QAAgB,CAAC,IAAI,KAAK,UAAU;YACxD,CAAC,CAAC,MAAO,QAAgB,CAAC,IAAI,CAAC,KAAK,CAAC;YACrC,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ;gBACzB,CAAC,CAAC,MAAM,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC;gBAC5B,CAAC,CAAC,IAAI,QAAQ,CAAC,KAAqB,CAAC,CAAA;QAEzC,iCAAiC;QACjC,sDAAsD;QACtD,kEAAkE;QAClE,sFAAsF;QACtF,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAA;QAChE,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,gCAAgC,CAAC,CAAA;QAE7D,kCAAkC;QAClC,+CAA+C;QAC/C,wFAAwF;QACxF,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE;QAC9C,+BAA+B;QAC/B,8BAA8B;SAC/B,CAAC,CAAA;QAEF,gBAAgB;QAChB,4BAA4B;QAC5B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAA;QACnE,+BAA+B;QAE/B,yCAAyC;QACzC,+EAA+E;QAC/E,sCAAsC;QACtC,MAAM,WAAW,GAAG,aAAa,CAAC,WAAW,EAAE,IAAI,CAAC,CAAA;QACpD,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,sBAAsB,CAAC,CAAA;QAEpD,0CAA0C;QAC1C,kCAAkC;QAClC,yCAAyC;QACzC,MAAM,UAAU,GAAG,MAAM,mBAAmB,CAAC,WAAW,EAAE,UAAU,CAAC,CAAA;QACrE,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,UAAU,EAAE,WAAW,CAAC,CAAA;QAEvD,OAAO;YACL,UAAU;YACV,iBAAiB,EAAE,UAAU,CAAC,KAAK;YACnC,kBAAkB,EAAE,UAAU,CAAC,MAAM;SACtC,CAAA;IACH,CAAC;CACF;AAED,SAAS,kBAAkB,CAAC,KAAmB,EAAE,EAAE,OAAO,KAA2B,EAAE;IACrF,IAAI,KAAK,GAAG,KAAK,CAAC,KAAK,CAAA;IACvB,IAAI,MAAM,GAAG,KAAK,CAAC,MAAM,CAAA;IACzB,IAAI,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,OAAO,EAAE,CAAC;QACjD,MAAM,KAAK,GAAG,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,OAAO,GAAG,MAAM,CAAA;QACjE,KAAK,GAAG,KAAK,GAAG,KAAK,CAAA;QACrB,MAAM,GAAG,MAAM,GAAG,KAAK,CAAA;IACzB,CAAC;IACD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG;IACvB,aAAa;IACb,YAAY;IACZ,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC,GAAG,SAAS,EACxC,SAAS,CACV,CAAA;IACD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,GAAG,SAAS,EAAE,SAAS,CAAC,CAAA;IAChF,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,CAAA;AAC/C,CAAC;AAED,SAAS,aAAa,CAAC,MAAc,EAAE,SAAiB;IACtD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IAC5B,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,KAAK,GAAG,MAAM,GAAG,CAAC,CAAC,CAAA;IAC/C,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;QACzD,MAAM,CAAC,GAAG,QAAQ,GAAG,CAAC,CAAA;QACtB,MAAM,KAAK,GAAI,QAAmB,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;QACxD,IAAI,CAAC,CAAC,CAAC,GAAG,KAAK,CAAA,CAAC,IAAI;QACpB,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAA,CAAC,IAAI;QACxB,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAA,CAAC,IAAI;QACxB,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAA,CAAC,IAAI;IACxB,CAAC;IACD,OAAO,IAAI,QAAQ,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAA;AAC9C,CAAC"}

package/build/models/ModelBase.d.ts

@@ -0,0 +1,14 @@
+import { type InferenceSession as InferenceSessionCommon, Tensor } from 'onnxruntime-common';
+import type { ImageRaw, LineImage, ModelBaseConstructorArg, ModelBaseOptions, ModelData, ReshapeOptions } from '../types/index.js';
+export declare class ModelBase {
+    #private;
+    options: ModelBaseOptions;
+    constructor({ model, options }: ModelBaseConstructorArg);
+    runModel({ modelData, onnxOptions, }: {
+        modelData: ModelData;
+        onnxOptions?: InferenceSessionCommon.RunOptions;
+    }): Promise<Tensor>;
+    imageToInput(image: ImageRaw, { mean, std }: ReshapeOptions): ModelData;
+    debugImage(image: ImageRaw | any, path: string): void;
+    debugBoxImage(sourceImage: ImageRaw | any, lineImages: LineImage[], path: string): Promise<void>;
+}

package/build/models/ModelBase.js

@@ -0,0 +1,53 @@
+import { Tensor } from 'onnxruntime-common';
+export class ModelBase {
+    options;
+    #model;
+    constructor({ model, options }) {
+        this.#model = model;
+        this.options = options;
+    }
+    async runModel({ modelData, onnxOptions = {}, }) {
+        const input = this.#prepareInput(modelData);
+        const outputs = await this.#model.run({
+            [this.#model.inputNames[0]]: input,
+        }, onnxOptions);
+        const output = outputs[this.#model.outputNames[0]];
+        return output;
+    }
+    #prepareInput(modelData) {
+        const input = Float32Array.from(modelData.data);
+        return new Tensor('float32', input, [1, 3, modelData.height, modelData.width]);
+    }
+    imageToInput(image, { mean = [0, 0, 0], std = [1, 1, 1] }) {
+        const R = [];
+        const G = [];
+        const B = [];
+        for (let i = 0; i < image.data.length; i += 4) {
+            R.push((image.data[i] / 255 - mean[0]) / std[0]);
+            G.push((image.data[i + 1] / 255 - mean[1]) / std[1]);
+            B.push((image.data[i + 2] / 255 - mean[2]) / std[2]);
+        }
+        const newData = [...B, ...G, ...R];
+        return {
+            data: newData,
+            width: image.width,
+            height: image.height,
+        };
+    }
+    debugImage(image, path) {
+        const { debugOutputDir, isDebug } = this.options;
+        if (!isDebug || !debugOutputDir) {
+            return;
+        }
+        image.write(`${debugOutputDir}/${path}`);
+    }
+    async debugBoxImage(sourceImage, lineImages, path) {
+        const { debugOutputDir, isDebug } = this.options;
+        if (!isDebug || !debugOutputDir) {
+            return;
+        }
+        const boxImage = await sourceImage.drawBox(lineImages);
+        boxImage.write(`${debugOutputDir}/${path}`);
+    }
+}
+//# sourceMappingURL=ModelBase.js.map

package/build/models/ModelBase.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ModelBase.js","sourceRoot":"","sources":["../../src/models/ModelBase.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmD,MAAM,EAAE,MAAM,oBAAoB,CAAA;AAW5F,MAAM,OAAO,SAAS;IACpB,OAAO,CAAkB;IACzB,MAAM,CAAkB;IAExB,YAAY,EAAE,KAAK,EAAE,OAAO,EAA2B;QACrD,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;QACnB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAA;IACxB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EACb,SAAS,EACT,WAAW,GAAG,EAAE,GAC0D;QAC1E,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAA;QAC3C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CACnC;YACE,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK;SACnC,EACD,WAAW,CACZ,CAAA;QACD,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAA;QAClD,OAAO,MAAM,CAAA;IACf,CAAC;IAED,aAAa,CAAC,SAAoB;QAChC,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;QAC/C,OAAO,IAAI,MAAM,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC,CAAA;IAChF,CAAC;IAED,YAAY,CAAC,KAAe,EAAE,EAAE,IAAI,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAkB;QACjF,MAAM,CAAC,GAAa,EAAE,CAAA;QACtB,MAAM,CAAC,GAAa,EAAE,CAAA;QACtB,MAAM,CAAC,GAAa,EAAE,CAAA;QACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9C,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;YAChD,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;YACpD,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;QACtD,CAAC;QACD,MAAM,OAAO,GAAG,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC,CAAA;QAClC,OAAO;YACL,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,MAAM,EAAE,KAAK,CAAC,MAAM;SACrB,CAAA;IACH,CAAC;IAED,UAAU,CAAC,KAAqB,EAAE,IAAY;QAC5C,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAA;QAChD,IAAI,CAAC,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC;YAChC,OAAM;QACR,CAAC;QACD,KAAK,CAAC,KAAK,CAAC,GAAG,cAAc,IAAI,IAAI,EAAE,CAAC,CAAA;IAC1C,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,WAA2B,EAAE,UAAuB,EAAE,IAAY;QACpF,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAA;QAChD,IAAI,CAAC,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC;YAChC,OAAM;QACR,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,UAAU,CAAC,CAAA;QACtD,QAAQ,CAAC,KAAK,CAAC,GAAG,cAAc,IAAI,IAAI,EAAE,CAAC,CAAA;IAC7C,CAAC;CACF"}

package/build/models/Recognition.d.ts

@@ -0,0 +1,12 @@
+import type { InferenceSession as InferenceSessionCommon, Tensor } from 'onnxruntime-common';
+import type { Dictionary, Line, LineImage, ModelBaseConstructorArg, ModelCreateOptions } from '../types/index.js';
+import { ModelBase } from './ModelBase.js';
+export declare class Recognition extends ModelBase {
+    #private;
+    static create({ models, onnxOptions, ...restOptions }: ModelCreateOptions): Promise<Recognition>;
+    constructor(options: ModelBaseConstructorArg, dictionary: Dictionary);
+    run(lineImages: LineImage[], { onnxOptions }?: {
+        onnxOptions?: InferenceSessionCommon.RunOptions;
+    }): Promise<Line[]>;
+    decodeText(output: Tensor): Line[];
+}