feat: add document OCR and offline model archives

2026-08-11 16:49:51 +08:00
parent 19a4469561
commit e0e7bc573c
46 changed files with 7831 additions and 85 deletions
@@ -0,0 +1,204 @@
+import {
+  documentOcrModelCatalogEntrySchema,
+  type DocumentOcrModelCatalogEntry
+} from '../shared/document-parsing-contracts'
+
+const detectionRevision =
+  '7d7f5d128d9309ebf6de4f21f404dd583afdbae3'
+const recognitionRevision =
+  'afba04b618200c5f4824531c6e42c957c6439d9a'
+const smallDetectionRevision =
+  '956a0b620a4017cc04056c692be1703b0025d028'
+const smallRecognitionRevision =
+  '296d43bc0ebced0fd9c605174aa5962e49810ab6'
+const mediumDetectionRevision =
+  'c317b40325be40bfaaff58c8dcece2a075294f8a'
+const mediumRecognitionRevision =
+  'db5d610d492a14e3c34dc1fd4e9339bd369f79e6'
+
+export const DOCUMENT_OCR_MODEL_CATALOG: readonly DocumentOcrModelCatalogEntry[] =
+  documentOcrModelCatalogEntrySchema.array().parse([
+    {
+      id: 'pp-ocrv6-tiny',
+      displayName: 'PP-OCRv6 Tiny',
+      description:
+        'PaddleOCR 官方轻量中文 OCR 模型，适合扫描 PDF 和图片的本地 CPU 识别。',
+      languages: ['中文', '英语'],
+      runtime: 'onnxruntime-web-wasm',
+      quality: 'basic',
+      speed: 'fast',
+      recommended: false,
+      repositoryUrl:
+        'https://modelscope.cn/models/PaddlePaddle/' +
+        'PP-OCRv6_tiny_rec_onnx',
+      license: {
+        name: 'Apache License 2.0',
+        notice:
+          '检测与识别模型由 PaddlePaddle 在 ModelScope 发布，使用前请阅读模型仓库及 PaddleOCR 的许可证说明。',
+        url: 'https://github.com/PaddlePaddle/PaddleOCR/blob/main/LICENSE'
+      },
+      files: [
+        {
+          name: 'detection.onnx',
+          role: 'detection',
+          download: {
+            url:
+              'https://modelscope.cn/models/PaddlePaddle/' +
+              'PP-OCRv6_tiny_det_onnx/resolve/' +
+              `${detectionRevision}/inference.onnx`,
+            size: 1_780_590,
+            sha256:
+              '193bab7a04fca699a6c82e6abb5b81bdb28177f0abd4062552b04908dafb19f8'
+          }
+        },
+        {
+          name: 'recognition.onnx',
+          role: 'recognition',
+          download: {
+            url:
+              'https://modelscope.cn/models/PaddlePaddle/' +
+              'PP-OCRv6_tiny_rec_onnx/resolve/' +
+              `${recognitionRevision}/inference.onnx`,
+            size: 4_462_639,
+            sha256:
+              '9ef676d6ed3c88256a2d92c640c44f25b0c40947e111b14b8be8f594091563e6'
+          }
+        },
+        {
+          name: 'dictionary.yml',
+          role: 'dictionary',
+          download: {
+            url:
+              'https://modelscope.cn/models/PaddlePaddle/' +
+              'PP-OCRv6_tiny_rec_onnx/resolve/' +
+              `${recognitionRevision}/inference.yml`,
+            size: 55_571,
+            sha256:
+              '66170210bad538e83fff3c4a3867e547d6bf20b50d64b20347c4b913f3034ea1'
+          }
+        }
+      ]
+    },
+    {
+      id: 'pp-ocrv6-small',
+      displayName: 'PP-OCRv6 Small',
+      description:
+        'PaddleOCR 官方 50 语言 OCR 模型，在识别质量、速度和本地资源占用之间取得平衡。',
+      languages: ['50 种语言'],
+      runtime: 'onnxruntime-web-wasm',
+      quality: 'balanced',
+      speed: 'balanced',
+      recommended: true,
+      repositoryUrl:
+        'https://modelscope.cn/models/PaddlePaddle/' +
+        'PP-OCRv6_small_rec_onnx',
+      license: {
+        name: 'Apache License 2.0',
+        notice:
+          '检测与识别模型由 PaddlePaddle 在 ModelScope 发布，使用前请阅读模型仓库及 PaddleOCR 的许可证说明。',
+        url: 'https://github.com/PaddlePaddle/PaddleOCR/blob/main/LICENSE'
+      },
+      files: [
+        {
+          name: 'detection.onnx',
+          role: 'detection',
+          download: {
+            url:
+              'https://modelscope.cn/models/PaddlePaddle/' +
+              'PP-OCRv6_small_det_onnx/resolve/' +
+              `${smallDetectionRevision}/inference.onnx`,
+            size: 9_880_512,
+            sha256:
+              'd73e0058b7a8086bbd57f3d10b8bcd4ff95363f67e06e2762b5e814fe9c9410e'
+          }
+        },
+        {
+          name: 'recognition.onnx',
+          role: 'recognition',
+          download: {
+            url:
+              'https://modelscope.cn/models/PaddlePaddle/' +
+              'PP-OCRv6_small_rec_onnx/resolve/' +
+              `${smallRecognitionRevision}/inference.onnx`,
+            size: 21_159_378,
+            sha256:
+              '5435fd747c9e0efe15a96d0b378d5bd157e9492ed8fd80edf08f30d02fa24634'
+          }
+        },
+        {
+          name: 'dictionary.yml',
+          role: 'dictionary',
+          download: {
+            url:
+              'https://modelscope.cn/models/PaddlePaddle/' +
+              'PP-OCRv6_small_rec_onnx/resolve/' +
+              `${smallRecognitionRevision}/inference.yml`,
+            size: 150_579,
+            sha256:
+              'ab078671bb49f06228eadccd34f1bb501e157f7a047095ffb943ba81512c77d1'
+          }
+        }
+      ]
+    },
+    {
+      id: 'pp-ocrv6-medium',
+      displayName: 'PP-OCRv6 Medium',
+      description:
+        'PaddleOCR 官方 50 语言高质量 OCR 模型，识别较慢，并需要更多内存且具有更高延迟。',
+      languages: ['50 种语言'],
+      runtime: 'onnxruntime-web-wasm',
+      quality: 'high',
+      speed: 'slow',
+      recommended: false,
+      repositoryUrl:
+        'https://modelscope.cn/models/PaddlePaddle/' +
+        'PP-OCRv6_medium_rec_onnx',
+      license: {
+        name: 'Apache License 2.0',
+        notice:
+          '检测与识别模型由 PaddlePaddle 在 ModelScope 发布，使用前请阅读模型仓库及 PaddleOCR 的许可证说明。',
+        url: 'https://github.com/PaddlePaddle/PaddleOCR/blob/main/LICENSE'
+      },
+      files: [
+        {
+          name: 'detection.onnx',
+          role: 'detection',
+          download: {
+            url:
+              'https://modelscope.cn/models/PaddlePaddle/' +
+              'PP-OCRv6_medium_det_onnx/resolve/' +
+              `${mediumDetectionRevision}/inference.onnx`,
+            size: 62_032_837,
+            sha256:
+              'eb13b44b25bb36f89528b68720af8a61d9cf381176107f465db1757b65d086e1'
+          }
+        },
+        {
+          name: 'recognition.onnx',
+          role: 'recognition',
+          download: {
+            url:
+              'https://modelscope.cn/models/PaddlePaddle/' +
+              'PP-OCRv6_medium_rec_onnx/resolve/' +
+              `${mediumRecognitionRevision}/inference.onnx`,
+            size: 76_554_979,
+            sha256:
+              '9c09abf0957f7968c7586464b7397b84ad2387a0497a351af40e9acc71b673ba'
+          }
+        },
+        {
+          name: 'dictionary.yml',
+          role: 'dictionary',
+          download: {
+            url:
+              'https://modelscope.cn/models/PaddlePaddle/' +
+              'PP-OCRv6_medium_rec_onnx/resolve/' +
+              `${mediumRecognitionRevision}/inference.yml`,
+            size: 150_580,
+            sha256:
+              '991b700facf5b50a7de193468207d5f4255b538dde0d312ae3b7c7a9b6873129'
+          }
+        }
+      ]
+    }
+  ])