CNWei
7d159c5702
- 在 ddddocr-core 中定义 ModelBuilder Trait 及其错误类型 - ddddocr-ort 支持 use_gpu、device_id 及 num_threads 链式配置与 CUDA 硬件加速 - ddddocr-tract 基于 multithread-mm 特性支持 CPU 线程数控制 - 支持基于 tract-linalg 配置推理线程数，显式引入 tract-linalg 的 multithread-mm 特性，控制 GEMM 算子并发 - 优化线程池加载策略，适配 Tokio 异步及 CLI 等多场景
23 lines
684 B
TOML
23 lines
684 B
TOML
[package]
|
|
name = "ddddocr-ort"
|
|
version.workspace = true
|
|
edition.workspace = true
|
|
license.workspace = true
|
|
|
|
[dependencies]
|
|
ddddocr-core = { path = "../ddddocr-core" } # 引入兄弟库
|
|
ort = { workspace = true ,features = ["cuda"]}
|
|
ndarray = { workspace = true } # 继承自工作空间
|
|
|
|
image = { workspace = true }
|
|
|
|
serde = { workspace = true }
|
|
serde_json = { workspace = true }
|
|
|
|
anyhow = { workspace = true }
|
|
thiserror = { workspace = true } # 刚好可以开始接入你需要的标准库错误处理
|
|
|
|
[features]
|
|
# 1. 声明 cuda feature
|
|
# 2. 当开启 ddddocr-ort 的 cuda feature 时，自动开启底层 ort 库的 cuda 支持（如果 ort 库支持的话）
|
|
cuda = ["ort/cuda"] |