@areb0s/ocr-common 1.0.0

package/src/types/types.ts

@@ -0,0 +1,90 @@
+import { InferenceSession } from 'onnxruntime-common'
+import { ImageRawBase as ImageRaw } from '#common/backend/ImageRawBase'
+import type { splitIntoLineImages } from '#common/backend/splitIntoLineImages'
+
+export { FileUtilsBase as FileUtils } from '#common/backend/FileUtilsBase'
+export { ImageRaw, InferenceSession }
+
+export type SplitIntoLineImages = typeof splitIntoLineImages
+
+export type ReshapeOptions = {
+  mean?: number[]
+  std?: number[]
+}
+
+export type ImageRawData = {
+  data: Uint8Array | Uint8ClampedArray // Uint8Array: Node Buffer, Uint8ClampedArray: Web ImageData
+  width: number
+  height: number
+}
+
+/**
+ * Browser-specific image input types
+ * Supports various browser image sources that can be converted to ImageRawData
+ */
+export type BrowserImageInput =
+  | string // URL or data URL
+  | ImageRawData // Raw pixel data
+  | ImageBitmap // From createImageBitmap(), OffscreenCanvas, etc.
+  | HTMLImageElement // <img> element
+  | HTMLCanvasElement // <canvas> element
+  | HTMLVideoElement // <video> element (captures current frame)
+
+export type ModelData = {
+  data: number[] | Uint8Array
+  width: number
+  height: number
+}
+
+export type Size = {
+  width: number
+  height: number
+}
+
+export type SizeOption = {
+  width?: number
+  height?: number
+  fit?: 'contain' | 'cover' | 'fill' | 'inside' | 'outside'
+}
+
+export type LineImage = {
+  image: ImageRaw
+  // TODO: [top, right, bottom, left], top: [x,y]
+  box: number[][]
+}
+
+export type Region = {
+  left: number
+  top: number
+  width: number
+  height: number
+}
+
+export type Line = {
+  text: string
+  mean: number
+  box?: number[][]
+}
+
+export type Dictionary = string[]
+
+export type Point = [x: number, y: number]
+
+export interface ModelBaseConstructorArg {
+  model: InferenceSession
+  options: ModelBaseOptions
+}
+
+export interface ModelBaseOptions {
+  isDebug?: boolean
+  debugOutputDir?: string
+}
+
+export interface ModelCreateOptions extends ModelBaseOptions {
+  models?: {
+    detectionPath: string
+    recognitionPath: string
+    dictionaryPath: string
+  }
+  onnxOptions?: InferenceSession.SessionOptions
+}

package/tsconfig.json

@@ -0,0 +1,10 @@
+{
+  "extends": "../../tsconfig.json",
+  // For publish
+  "compilerOptions": {
+    "baseUrl": ".",
+    "paths": {
+      "#common/*": ["./src/*"]
+    }
+  }
+}