# RapidOCR 参数详解(结合本项目 RapidOCR 3.8.1) > 本文面向 `D:\PycharmProjects\ocr-server` 的实际使用与调优。 > > 核验环境(2026-07-17): > > - `rapidocr==3.8.1` > - `onnxruntime-directml==1.23.0` > - 项目入口:`engine/ocr_engine.py` > - 本项目目前使用整图 OCR,不再使用分块模式 > > **重要:** RapidOCR 官方 `main/latest` 文档目前主要描述 3.9.x。3.9.0 起默认模型和部分枚举发生变化,不能直接把新文档中的所有默认值套到本项目 3.8.1。 --- ## 1. 先理解 RapidOCR 的参数层级 RapidOCR 参数分为三层: 1. **构造配置**:创建 `RapidOCR` 实例时确定模型、推理后端、预处理及默认行为。 2. **Det / Cls / Rec 模块配置**:分别控制检测、方向分类、文字识别。 3. **运行时调用参数**:调用 `engine(image, ...)` 时覆盖少量可动态调整的选项。 配置树的顶层结构是: ```yaml Global: # OCR 流程和输出的全局设置 EngineConfig: # 各推理后端设置 Det: # 文本检测 Cls: # 文本行方向分类 Rec: # 文本识别 ``` 参数路径使用点号表达,例如: ```python params = { "Global.text_score": 0.6, "Det.box_thresh": 0.55, "EngineConfig.onnxruntime.use_dml": True, } ``` ### 1.1 三种创建方式 #### 使用包内默认配置 ```python from rapidocr import RapidOCR engine = RapidOCR() ``` #### 使用完整 YAML ```python engine = RapidOCR(config_path="rapidocr.yaml") ``` YAML 中枚举参数写字符串: ```yaml Det: engine_type: onnxruntime lang_type: ch model_type: mobile ocr_version: PP-OCRv4 ``` #### 使用 `params` 字典覆盖 ```python from rapidocr import EngineType, LangDet, ModelType, OCRVersion, RapidOCR engine = RapidOCR(params={ "Det.engine_type": EngineType.ONNXRUNTIME, "Det.lang_type": LangDet.CH, "Det.model_type": ModelType.MOBILE, "Det.ocr_version": OCRVersion.PPOCRV4, "Det.box_thresh": 0.55, }) ``` **3.8.1 的重要规则:** `params` 字典中的枚举字段必须传 RapidOCR 的 Enum,不能传普通字符串。包括: - `engine_type` - `model_type` - `ocr_version` - `lang_type` - `task_type` 例如以下写法在 `params` 中可能触发 `TypeError`: ```python # 不适合 rapidocr 3.8.1 的 params 字典 {"Det.engine_type": "onnxruntime"} ``` 但同一个值在 YAML 中应当写字符串,因为加载 YAML 后 RapidOCR 会自动将其转换成 Enum。 ### 1.2 配置覆盖顺序 构造时的实际顺序为: 1. 如果 `config_path` 存在,加载该 YAML; 2. 否则加载 RapidOCR 包内 `config.yaml`; 3. 最后用 `params` 逐项覆盖。 因此: ```python RapidOCR(config_path="a.yaml", params={"Global.text_score": 0.7}) ``` 表示先加载 `a.yaml`,再把其中的 `Global.text_score` 改为 `0.7`。 如果 `config_path` 指向不存在的文件,3.8.1 源码会退回包内默认配置,而不是必然报错。生产环境应主动检查路径,避免拼错路径后静默使用默认配置。 --- ## 2. OCR 三阶段是什么 完整 OCR 流程通常是: ```text 输入图片 ↓ Det:检测文字区域 ↓ 裁剪每个文本行 ↓ Cls:判断文本行是否需要旋转 180° ↓ Rec:识别文本内容 ↓ 按置信度过滤并输出 ``` 三个开关的典型组合: | 场景 | use_det | use_cls | use_rec | 返回类型/意义 | |---|---:|---:|---:|---| | 完整 OCR | true | 按需 | true | `RapidOCROutput` | | 只检测文字框 | true | false | false | `TextDetOutput` | | 已裁切文本行,只识别 | false | false | true | `TextRecOutput` | | 已裁切文本行,方向分类后识别 | false | true | true | 无检测框,处理输入整体 | | 只做方向分类 | false | true | false | `TextClsOutput` | 注意:如果关闭 Det,却把一张包含很多文本行的整页图交给 Rec,Rec 会把整张图当成一条待识别文本行,通常不是想要的效果。 --- ## 3. Global 全局参数 以下默认值以本项目安装的 **RapidOCR 3.8.1 包内 `config.yaml`** 为准,而不是当前 3.9.x 在线文档。 | 参数 | 3.8.1 默认值 | 作用 | 调优建议 | |---|---:|---|---| | `text_score` | `0.5` | 最终识别置信度过滤阈值 | 误识别太多时提高;漏掉低置信度文字时降低 | | `use_det` | `true` | 是否执行文本检测 | 整页图片一般开启 | | `use_cls` | `true` | 是否执行 0°/180°方向分类 | 方向固定时可关闭以降低耗时 | | `use_rec` | `true` | 是否执行文字识别 | 只要框时可关闭 | | `min_height` | `30` | 全局预处理的小尺寸/竖图补边相关阈值 | 通常保留模型默认 | | `width_height_ratio` | `8` | 竖长图触发补边逻辑的宽高比阈值 | 通常不要作为第一调优项 | | `max_side_len` | `2000` | 整体输入预处理允许的最大边 | 大图精度与内存/耗时的重要平衡项 | | `min_side_len` | `30` | 整体输入预处理允许的最小边 | 极小图片可能被放大 | | `return_word_box` | `false` | 是否计算字/词级坐标 | 需要细粒度坐标时开启,会增加后处理工作 | | `return_single_char_box` | `false` | 是否将英文数字进一步拆到单字符坐标 | 仅在 `return_word_box=True` 时有意义 | | `font_path` | `null` | 可视化结果使用的字体 | 只影响可视化,不改善识别准确率 | | `log_level` | `info` | RapidOCR 日志级别 | 排查后端和模型加载时可设 `debug` | | `model_root_dir` | `null` | 内置/自动下载模型的根目录 | `null` 时使用包目录下 `models` | ### 3.1 `text_score` 与 `box_thresh` 不一样 这是最常混淆的一组参数: - `Det.box_thresh`:检测阶段判断一个候选区域是否像文本。 - `Global.text_score`:识别完成后,按识别置信度过滤结果。 例子: - 图片中某行完全没有检测框:先检查 `limit_side_len`、`thresh`、`box_thresh`。 - 框已经检测到了,但最终结果中没有这行:可能是 `text_score` 太高,或者 Rec 识别失败。 ### 3.2 `max_side_len` 与 `Det.limit_side_len` 不一样 - `Global.max_side_len`:进入整个 OCR 流程前,对原图做一次全局尺寸约束;3.8.1 默认 `2000`。 - `Det.limit_side_len`:检测模型自身预处理的缩放目标;3.8.1 默认 `736`。 两者可能连续生效。把 `Det.limit_side_len` 调得很大,如果图片之前已被 `Global.max_side_len` 缩小,也无法恢复被缩掉的细节。 调大图小字时,应先确认这两个层级,而不是只调整其中一个。 ### 3.3 字/词/字符坐标 `return_word_box=True` 时,RapidOCR 会结合识别结果和文本行框计算更细粒度坐标。其“word”的粒度与语言、版本有关,不应简单理解成永远返回英文意义上的单词: - 中文及中英混合通常更接近按字拆分; - 纯英文通常更接近按词拆分; - `return_single_char_box=True` 用于进一步获取英文/数字字符级坐标,而且依赖 `return_word_box=True`。 本项目每次调用都显式传入: ```python return_word_box=True ``` 因此构造配置中的 `Global.return_word_box` 会在首次项目调用时被覆盖为 `True`。 --- ## 4. Det 文本检测参数 Det 的目标是找到文字区域,输出四点框。 ### 4.1 模型选择参数 | 参数 | 3.8.1 默认值 | 作用 | |---|---|---| | `engine_type` | `onnxruntime` | Det 使用的推理后端 | | `lang_type` | `ch` | 检测模型语言类别 | | `model_type` | `mobile` | 模型规模,本地 3.8.1 只有 `mobile/server` 枚举 | | `ocr_version` | `PP-OCRv4` | OCR 模型系列 | | `task_type` | `det` | 模型任务类型,通常不要修改 | | `model_path` | `null` | 明确指定模型文件 | | `model_dir` | `null` | 部分后端使用的模型目录 | 如果 `model_path` 不为空,ONNXRuntime 会直接使用该文件;模型选择元数据主要用于 `model_path=null` 时定位 RapidOCR 托管的默认模型。 ### 4.2 检测预处理与后处理参数 | 参数 | 3.8.1 默认值 | 含义 | 调大/调小的典型影响 | |---|---:|---|---| | `limit_side_len` | `736` | 检测前图像缩放边长限制 | 调大通常保留更多小字细节,但更慢、更耗显存/内存 | | `limit_type` | `min` | `limit_side_len` 作用方式 | 与模型训练预处理有关,不建议随意改 | | `std` | `[0.5,0.5,0.5]` | 输入归一化标准差 | 必须匹配模型训练配置 | | `mean` | `[0.5,0.5,0.5]` | 输入归一化均值 | 必须匹配模型训练配置 | | `thresh` | `0.3` | DB 像素二值化阈值 | 降低会产生更多候选区域,也可能增加噪声 | | `box_thresh` | `0.5` | 候选文本框平均得分阈值 | 降低增加召回;提高减少假框 | | `max_candidates` | `1000` | 最多处理候选区域数 | 超密集页面可能受限;调大增加后处理成本 | | `unclip_ratio` | `1.6` | 检测框向外扩张比例 | 调大可避免切掉字符边缘,但可能框入邻行 | | `use_dilation` | `true` | 二值图是否做膨胀 | 有助于连接断裂区域,也可能粘连相邻文字 | | `score_mode` | `fast` | 文本框评分方式 | `fast` 优先速度;其他值须以版本源码支持为准 | ### 4.3 检测调优顺序 #### 小字检测不到 建议按以下顺序检查: 1. 原始图片是否已被业务预处理缩小; 2. `Global.max_side_len` 是否过小; 3. 适当提高 `Det.limit_side_len`; 4. 小幅降低 `Det.box_thresh`; 5. 再考虑小幅降低 `Det.thresh`。 不要一开始就同时大幅降低两个阈值,否则很难判断是哪项有效,而且容易产生大量背景假框。 #### 框太紧、字符边缘被切掉 优先小幅提高: ```yaml Det: unclip_ratio: 1.8 ``` 框入邻行时反向降低。 #### 假框很多 优先提高: ```yaml Det: box_thresh: 0.6 ``` 若大量纹理像素形成候选,再提高 `thresh`。 --- ## 5. Cls 文本行方向分类参数 Cls 通常只判断裁切文本行是 `0°` 还是 `180°`。它不是通用的任意角度纠偏模块,不能替代 90°旋转、透视矫正或倾斜校正。 | 参数 | 3.8.1 默认值 | 作用 | |---|---:|---| | `engine_type` | `onnxruntime` | 分类推理后端 | | `lang_type` | `ch` | 3.8.1 的 `LangCls` 只支持 `ch` | | `model_type` | `mobile` | 模型规格 | | `ocr_version` | `PP-OCRv4` | 模型系列 | | `task_type` | `cls` | 固定为分类任务 | | `model_path` | `null` | 自定义分类模型路径 | | `cls_image_shape` | `[3,48,192]` | 分类输入形状 | | `cls_batch_num` | `6` | 一次分类的文本行数量 | | `cls_thresh` | `0.9` | 达到该置信度才按分类结果旋转 | | `label_list` | `["0","180"]` | 分类标签 | 调优建议: - 扫描件方向稳定:关闭 `use_cls`,通常能降低延迟。 - 确实混有上下颠倒文本:开启 `use_cls`。 - 不要期待 Cls 修正任意角度倾斜。 - `cls_batch_num` 增大可能提高大量文本行时的吞吐,但会增加瞬时内存/显存;应实测。 - `cls_image_shape`、`label_list` 必须与模型匹配,不是通用性能旋钮。 本项目 `.env` 当前为: ```env OCR_USE_CLS=true ``` 所以会运行分类阶段。 --- ## 6. Rec 文字识别参数 Rec 接收裁切后的文本行,输出文字和识别置信度。 | 参数 | 3.8.1 默认值 | 作用 | |---|---:|---| | `engine_type` | `onnxruntime` | 识别推理后端 | | `lang_type` | `ch` | 字符语言/字典族 | | `model_type` | `mobile` | 模型规模 | | `ocr_version` | `PP-OCRv4` | OCR 模型系列 | | `task_type` | `rec` | 固定为识别任务 | | `model_path` | `null` | 自定义识别模型路径 | | `model_dir` | `null` | 部分后端模型目录 | | `rec_keys_path` | `null` | 自定义字符字典路径 | | `rec_img_shape` | `[3,48,320]` | 识别模型输入形状 | | `rec_batch_num` | `6` | 一次识别的文本行数量 | ### 6.1 `lang_type` 支持范围(本地 3.8.1) 本地 `LangRec` 枚举包括: ```text ch, ch_doc, en, arabic, chinese_cht, cyrillic, devanagari, japan, korean, ka, latin, ta, te, eslav, th, el ``` 模型文件、`lang_type` 和字符字典必须彼此匹配。 ### 6.2 `rec_keys_path` 不是“额外允许字符” 它是识别模型的字符索引映射。随意给一个新字典,并不会让原模型学会新字符,反而可能导致输出索引映射错误。 只有自定义模型确实按该字典训练/导出时才应设置。 ### 6.3 `rec_img_shape` 不应随意改 它必须符合模型输入和预处理约定。仅为了“识别更长文本”直接改宽度,不保证有效;动态宽度能力、模型导出方式和后端均会影响是否支持。 ### 6.4 `rec_batch_num` - 增大:可能提高多文本行吞吐,增加峰值内存/显存。 - 减小:降低峰值资源,可能降低吞吐。 - 对一张只有少量文本行的图片,增大批量不一定更快。 --- ## 7. EngineConfig 与推理后端 `Det.engine_type`、`Cls.engine_type`、`Rec.engine_type` 决定各阶段使用哪个后端;`EngineConfig.<后端>` 只是该后端的参数。 例如: ```python params = { "Det.engine_type": EngineType.ONNXRUNTIME, "Rec.engine_type": EngineType.ONNXRUNTIME, "EngineConfig.onnxruntime.use_dml": True, } ``` 如果把 `EngineConfig.tensorrt.use_fp16=True` 写好了,但三个模块仍是 `EngineType.ONNXRUNTIME`,TensorRT 参数不会被使用。 RapidOCR 支持 Det、Cls、Rec 分别使用不同后端,但混用会增加部署依赖、初始化复杂度和排障成本。一般先统一后端。 --- ## 8. ONNXRuntime 参数 ### 8.1 通用会话参数 | 参数 | 3.8.1 默认值 | 作用 | |---|---:|---| | `intra_op_num_threads` | `-1` | 单个算子内部并行线程数 | | `inter_op_num_threads` | `-1` | 算子之间并行线程数 | | `enable_cpu_mem_arena` | `false` | 是否启用 ORT CPU 内存池 | | `cpu_ep_cfg.arena_extend_strategy` | `kSameAsRequested` | CPU 内存池扩展策略 | 3.8.1 源码中: - `-1` 表示不显式设置,由 ONNXRuntime 处理; - 只有线程数位于 `1..os.cpu_count()` 时才写入 SessionOptions; - 超出 CPU 数或无效值会被跳过,不会按填写值生效。 Web 服务调优时不要只追求单请求最快: - 单进程、单请求:可适当提高 `intra_op_num_threads`; - 多个并发请求:每个 Session 使用过多线程会相互争抢,反而使尾延迟恶化; - `inter_op_num_threads` 并非越大越好,应以真实并发压测决定。 ### 8.2 CPU 配置示例 ```python params = { "Det.engine_type": EngineType.ONNXRUNTIME, "Cls.engine_type": EngineType.ONNXRUNTIME, "Rec.engine_type": EngineType.ONNXRUNTIME, "EngineConfig.onnxruntime.intra_op_num_threads": 4, "EngineConfig.onnxruntime.inter_op_num_threads": 2, } ``` 本项目 `OCR_DEVICE=cpu` 时正是这样构建参数。 --- ## 9. DirectML 参数 DirectML 不是独立的 `engine_type`;它是 ONNXRuntime 的 Execution Provider。 正确关系是: ```text Det/Cls/Rec.engine_type = onnxruntime EngineConfig.onnxruntime.use_dml = true 安装 onnxruntime-directml ``` 示例: ```python params = { "Det.engine_type": EngineType.ONNXRUNTIME, "Cls.engine_type": EngineType.ONNXRUNTIME, "Rec.engine_type": EngineType.ONNXRUNTIME, "EngineConfig.onnxruntime.use_dml": True, } ``` ### 9.1 生效条件 本地 3.8.1 源码会检查: 1. 操作系统必须是 Windows; 2. Windows Build 必须不低于 `18362`; 3. `onnxruntime.get_available_providers()` 中必须有 `DmlExecutionProvider`。 不满足时会记录警告,并回退到默认可用 Provider,通常是 CPU。 检查命令: ```bash .venv/Scripts/python.exe -c "import onnxruntime as ort; print(ort.get_available_providers())" ``` 预期包含: ```text DmlExecutionProvider CPUExecutionProvider ``` ### 9.2 `dm_ep_cfg` 拼写 RapidOCR 3.8.1 配置字段是: ```yaml EngineConfig: onnxruntime: use_dml: true dm_ep_cfg: null ``` 是 `dm_ep_cfg`,不是 `dml_ep_cfg`。 当它为 `null` 时,3.8.1 会复用 CPU Provider 配置;若 CUDA 也被认为可用,则可能复用 CUDA 配置。一般不需要手动填写 DirectML provider options。 ### 9.3 DirectML 常见误区 - 日志打印“推理设备: DirectML”只代表应用选择了 DML 配置,不单独证明实际 Session 使用 DML。 - 要结合可用 Provider 和 RapidOCR/ORT 日志确认。 - 同一环境不要混装多个互斥的 `onnxruntime`、`onnxruntime-gpu`、`onnxruntime-directml` 包,否则 Provider 和二进制版本可能相互覆盖。 - DirectML 首次初始化可能较慢;是否比 CPU 快必须用真实图片、完整三阶段和并发模型实测。 --- ## 10. ONNXRuntime CUDA 参数 CUDA 也是 ONNXRuntime Execution Provider,不是 `EngineType.CUDA`。 ```python params = { "Det.engine_type": EngineType.ONNXRUNTIME, "Cls.engine_type": EngineType.ONNXRUNTIME, "Rec.engine_type": EngineType.ONNXRUNTIME, "EngineConfig.onnxruntime.use_cuda": True, "EngineConfig.onnxruntime.cuda_ep_cfg.device_id": 0, } ``` | 参数 | 默认值 | 作用 | |---|---:|---| | `use_cuda` | `false` | 请求启用 CUDA EP | | `cuda_ep_cfg.device_id` | `0` | GPU 编号 | | `arena_extend_strategy` | `kNextPowerOfTwo` | GPU 内存池扩展策略 | | `cudnn_conv_algo_search` | `EXHAUSTIVE` | cuDNN 卷积算法搜索策略 | | `do_copy_in_default_stream` | `true` | 是否在默认流中执行复制 | 生效还需要: - 安装兼容的 `onnxruntime-gpu`; - CUDA/cuDNN 与 ORT 版本匹配; - `get_available_providers()` 包含 `CUDAExecutionProvider`。 仅设置 `use_cuda=True` 不会自动安装 CUDA,也不保证实际使用 GPU。条件不满足时,RapidOCR 3.8.1 会警告并回退。 --- ## 11. TensorRT 参数 TensorRT 在 RapidOCR 中是独立的推理引擎: ```python params = { "Det.engine_type": EngineType.TENSORRT, "Cls.engine_type": EngineType.TENSORRT, "Rec.engine_type": EngineType.TENSORRT, "EngineConfig.tensorrt.device_id": 0, "EngineConfig.tensorrt.use_fp16": True, } ``` | 参数 | 3.8.1 默认值 | 作用 | |---|---:|---| | `device_id` | `0` | GPU 编号 | | `use_fp16` | `true` | 使用 FP16 | | `use_int8` | `false` | 使用 INT8;通常需要正确量化支持 | | `workspace_size` | `1073741824` | TensorRT workspace,默认 1 GiB | | `cache_dir` | `null` | engine 缓存目录 | | `force_rebuild` | `false` | 是否强制重建 engine | | `det_profile` | 见默认配置 | 检测动态 shape 范围 | | `rec_profile` | 见默认配置 | 识别动态 shape 范围 | | `cls_profile` | 见默认配置 | 分类动态 shape 范围 | 默认动态 shape: ```yaml EngineConfig: tensorrt: det_profile: min_shape: [1, 3, 32, 32] opt_shape: [1, 3, 736, 736] max_shape: [1, 3, 2048, 2048] rec_profile: min_shape: [1, 3, 48, 32] opt_shape: [6, 3, 48, 320] max_shape: [6, 3, 48, 2048] cls_profile: min_shape: [1, 3, 48, 32] opt_shape: [6, 3, 48, 192] max_shape: [6, 3, 48, 192] ``` 注意: - 输入 shape 超出 profile 可能失败或触发重新构建; - 首次 ONNX 转 TensorRT engine 会明显较慢,后续使用缓存; - `.engine` 通常与 GPU 架构、TensorRT/CUDA 版本、模型和 profile 相关,不宜跨机器盲目复用; - `use_fp16=True` 并不保证所有算子都以 FP16 执行,也不保证所有 GPU 都同等受益; - TensorRT 部署依赖比 ONNXRuntime CPU/DirectML 高,先确认稳定性再追求吞吐。 --- ## 12. 其他后端参数概览 本地 3.8.1 的 `EngineType` 还包括: ```text openvino, paddle, torch, mnn ``` 配置树还有: - `EngineConfig.openvino` - `EngineConfig.paddle` - `EngineConfig.torch` - `EngineConfig.mnn` 但“配置中出现”不等于当前环境已安装对应依赖,也不等于任意模型组合都可运行。切换前应同时确认: 1. 本地 RapidOCR 版本支持该后端; 2. 后端 Python 包和原生运行库已安装; 3. 模型格式与该后端匹配; 4. 枚举、模型选择和配置项属于同一个版本。 --- ## 13. `RapidOCR.__call__()` 运行时参数 本地 RapidOCR 3.8.1 的实际签名: ```python engine( img_content, use_det=None, use_cls=None, use_rec=None, return_word_box=None, return_single_char_box=None, text_score=None, box_thresh=None, unclip_ratio=None, ) ``` ### 13.1 参数表 | 参数 | 作用 | |---|---| | `img_content` | 图片路径、URL(取决于加载器支持)、`numpy.ndarray`、bytes 或 `Path` | | `use_det` | 覆盖检测开关 | | `use_cls` | 覆盖方向分类开关 | | `use_rec` | 覆盖识别开关 | | `return_word_box` | 覆盖字/词级坐标开关 | | `return_single_char_box` | 覆盖英文数字单字符坐标开关 | | `text_score` | 覆盖最终识别置信度阈值 | | `box_thresh` | 直接更新 Det 后处理器的框阈值 | | `unclip_ratio` | 直接更新 Det 后处理器的扩框比例 | 其它构造参数(例如 `limit_side_len`、`rec_batch_num`、`engine_type`)不能直接作为本地 3.8.1 `__call__()` 的关键字参数传入。 ### 13.2 运行时“覆盖”会保留到后续调用 这是 3.8.1 源码里非常容易忽略的行为:`update_params()` 直接修改实例属性,而不是建立仅本次调用的临时配置。 ```python engine(img1, text_score=0.8) engine(img2) # text_score 仍然是 0.8,不会自动恢复构造值 ``` 传 `None` 表示“不修改当前状态”,不是“恢复默认值”。 因此共享同一个 `RapidOCR` 实例的 Web 服务,不应允许不同并发请求随意传不同的运行时阈值。否则一次请求可能修改后续请求看到的状态,并存在并发竞态。 推荐做法: - 服务启动时固定构造参数; - 业务请求只传固定的 `use_det/use_cls/use_rec/return_word_box`; - 如果必须支持不同配置,创建独立的引擎实例池,或者在外部做串行化与显式恢复。 ### 13.3 未知参数 `__call__()` 的 Python 签名不接受任意 `**kwargs`。传入未声明参数会由 Python 直接报 `TypeError`。 --- ## 14. 模型选择、模型路径与自动下载 ### 14.1 `model_path` 指定时 ONNXRuntime 直接加载该路径: ```python params = { "Det.model_path": "model/custom_det.onnx", } ``` 此时 RapidOCR 不会根据 `ocr_version/model_type/lang_type` 替你验证文件内容是否真的匹配这些声明。最终是否可用取决于模型输入输出、后处理和字典是否兼容。 ### 14.2 `model_path=null` 时 RapidOCR 会根据以下组合查找托管模型: ```text engine_type + ocr_version + task_type + lang_type + model_type ``` 若本地模型不存在,则按照内置模型清单下载并校验 SHA256。该能力只适用于 RapidOCR 已登记的托管模型组合,不表示任意自定义路径缺失时都会自动下载。 ### 14.3 `model_root_dir` `Global.model_root_dir=null` 时,本地 3.8.1 会设为 RapidOCR 安装目录下的 `models`。如果运行环境对 site-packages 没有写权限,而又需要下载非内置模型,应显式指定一个可写目录。 --- ## 15. 版本差异 ### 15.1 本项目:RapidOCR 3.8.1 本地枚举仅包含: ```python OCRVersion: PP-OCRv4, PP-OCRv5 ModelType: mobile, server ``` 包内无参默认配置是: - Det:PP-OCRv4 mobile + ONNXRuntime - Cls:PP-OCRv4 mobile + ONNXRuntime - Rec:PP-OCRv4 mobile + ONNXRuntime ### 15.2 RapidOCR 3.9.0 及以后 官方文档说明 3.9.0 起默认组合改为: - Det:PP-OCRv6 small - Cls:PP-OCRv4 mobile - Rec:PP-OCRv6 small - 默认后端仍是 ONNXRuntime 这意味着升级后,如果没有显式固定模型,默认模型可能改变,准确率、速度、模型文件和输出行为都可能变化。 **升级建议:** 1. 先生成目标版本默认配置; 2. 对照新版本 Enum 和参数树; 3. 明确固定 Det/Cls/Rec 模型; 4. 用真实业务样本比较准确率、耗时和资源占用; 5. 不要直接复制 3.9.x 示例到 3.8.1。 --- ## 16. 本项目当前实际配置 项目不维护 RapidOCR YAML 副本,唯一配置链是: ```text 项目根 .env → config/config.py::Settings → main.py → OCRProcessor → RapidOCR(params=...) ``` 修改 `.env` 后需要重启服务。项目开放以下常用参数: | `.env` | 默认值 | RapidOCR 映射/作用 | |---|---:|---| | `OCR_USE_DET` | `true` | 每次调用的 `use_det` | | `OCR_USE_CLS` | `false` | 每次调用的 `use_cls` | | `OCR_USE_REC` | `true` | 每次调用的 `use_rec` | | `OCR_MODEL_DIR` | `model` | 三个模型文件所在目录;相对项目根解析 | | `OCR_TEXT_SCORE` | `0.5` | `Global.text_score` | | `OCR_MAX_SIDE_LEN` | `2000` | `Global.max_side_len` | | `OCR_DET_LIMIT_SIDE_LEN` | `736` | `Det.limit_side_len` | | `OCR_DET_BOX_THRESH` | `0.5` | `Det.box_thresh` | | `OCR_REC_BATCH_NUM` | `6` | `Rec.rec_batch_num` | | `OCR_DEVICE` | `cpu` | `cpu/cuda/dml/tensorrt` | | `OCR_DEVICE_ID` | `0` | GPU 编号 | | `OCR_INTRA_THREADS` | `4` | ORT CPU 算子内线程数 | | `OCR_INTER_THREADS` | `2` | ORT CPU 算子间线程数 | 模型目录会在配置加载时转成绝对路径,`OCRProcessor` 启动时检查以下文件: ```text ch_PP-OCRv5_det_mobile.onnx ch_PP-LCNet_x0_25_textline_ori_cls_mobile.onnx ch_PP-OCRv5_rec_mobile.onnx ``` 三阶段模型契约固定为 PP-OCRv5 mobile: ```python Det: EngineType.ONNXRUNTIME, LangDet.CH, ModelType.MOBILE, OCRVersion.PPOCRV5 Cls: EngineType.ONNXRUNTIME, LangCls.CH, ModelType.MOBILE, OCRVersion.PPOCRV5 Rec: EngineType.ONNXRUNTIME, LangRec.CH, ModelType.MOBILE, OCRVersion.PPOCRV5 ``` 这些模型契约不是普通调优参数,因此不通过 `.env` 开放。其余本文介绍的底层参数用于理解 RapidOCR;除上表外,项目目前并未提供外部配置入口。 每次识别仍显式固定三阶段开关,并请求 `return_word_box=True`。阈值在引擎构造时设置,不按单个请求动态修改,避免共享实例状态串扰。 --- ## 17. 推荐配置示例 ### 17.1 本项目 DirectML 稳定起点 ```python params = { "Global.text_score": 0.5, "Global.max_side_len": 2000, "Det.model_path": "model/ch_PP-OCRv5_det_mobile.onnx", "Det.engine_type": EngineType.ONNXRUNTIME, "Det.lang_type": LangDet.CH, "Det.model_type": ModelType.MOBILE, "Det.ocr_version": OCRVersion.PPOCRV5, "Det.limit_side_len": 736, "Det.thresh": 0.3, "Det.box_thresh": 0.5, "Det.unclip_ratio": 1.6, "Cls.model_path": "model/ch_PP-LCNet_x0_25_textline_ori_cls_mobile.onnx", "Cls.engine_type": EngineType.ONNXRUNTIME, "Cls.model_type": ModelType.MOBILE, "Rec.model_path": "model/ch_PP-OCRv5_rec_mobile.onnx", "Rec.engine_type": EngineType.ONNXRUNTIME, "Rec.lang_type": LangRec.CH, "Rec.model_type": ModelType.MOBILE, "Rec.ocr_version": OCRVersion.PPOCRV5, "EngineConfig.onnxruntime.use_dml": True, } ``` > 上面是依据文件命名给出的统一方向,不代表已经验证这些模型的全部元数据。应用前应使用真实样本回归。 ### 17.2 CPU 并发服务起点 ```python params = { "EngineConfig.onnxruntime.intra_op_num_threads": 4, "EngineConfig.onnxruntime.inter_op_num_threads": 1, "Global.text_score": 0.5, "Det.limit_side_len": 736, "Rec.rec_batch_num": 6, } ``` 线程数必须按机器核心数和服务并发压测,不存在对所有机器都最优的固定值。 ### 17.3 小字优先配置 ```python params = { "Global.max_side_len": 3000, "Det.limit_side_len": 1280, "Det.box_thresh": 0.45, "Det.unclip_ratio": 1.7, } ``` 影响:通常更慢、更耗内存/显存,也可能增加假框。应逐项修改,并记录召回、误检和时延。 ### 17.4 减少误识别 ```python params = { "Det.box_thresh": 0.6, "Global.text_score": 0.65, } ``` - 提高 `box_thresh`:减少低质量检测框; - 提高 `text_score`:过滤低置信度识别结果。 两者同时提高会降低召回率,应以业务验收集评估。 ### 17.5 已裁切单行文本 ```python result = engine( line_image, use_det=False, use_cls=False, use_rec=True, return_word_box=False, ) ``` 这适合输入已经是单行文本的情况,不适合整页图片。 --- ## 18. 常见无效、冲突或误导性配置 ### 18.1 只设置 `use_dml/use_cuda`,但后端不是 ONNXRuntime ```python "Det.engine_type": EngineType.TENSORRT, "EngineConfig.onnxruntime.use_dml": True, ``` DML 参数对 TensorRT Det 无效。 ### 18.2 只写 GPU 参数,没有安装对应 Provider `use_cuda=True` 或 `use_dml=True` 只是请求使用该 Provider。Provider 不存在时会回退,必须检查: ```python onnxruntime.get_available_providers() ``` ### 18.3 同时打开多个 ONNXRuntime 加速 Provider 同时 `use_cuda=True` 和 `use_dml=True` 会形成 Provider 优先级,并不代表两张后端共同加速一个推理。3.8.1 的插入顺序还会影响谁排在前面。一般一次只启用一个主要 GPU Provider。 ### 18.4 显式 `model_path` 与模型声明不一致 文件可能仍能加载,但声明会误导下载逻辑、维护者和未来后端。应统一 `ocr_version/model_type/lang_type/task_type`。 ### 18.5 修改 `mean/std/image_shape/label_list` 来“调准确率” 这些通常是模型契约,不是普通阈值。与模型不匹配会直接破坏结果。 ### 18.6 只调 `Det.limit_side_len`,忽略 `Global.max_side_len` 原图可能已经先被全局缩放,检测阶段无法恢复细节。 ### 18.7 把 `text_score` 当检测阈值 它只过滤识别置信度。根本没有框时,降低 `text_score` 通常无效。 ### 18.8 把 Cls 当任意旋转矫正 默认标签只有 `0/180`,不处理所有角度。 ### 18.9 运行时参数被误认为只影响一次调用 本地 3.8.1 会把非 `None` 运行时值写回共享实例。Web 并发中动态阈值可能相互影响。 ### 18.10 依赖 `main/latest` 默认值但不锁定版本 RapidOCR 3.9.0 已更换默认 Det/Rec 模型。生产环境应锁定包版本和模型选择。 --- ## 19. 推荐调优方法 不要凭单张图片调参数。建议准备固定验收集,至少覆盖: - 正常扫描件; - 小字高分辨率图; - 模糊、压缩、低对比度图; - 横排、上下颠倒文本; - 中英数字混合; - 空白和复杂纹理背景。 记录指标: - 文本行召回率; - 误检框数量; - 字符/字段准确率; - 单请求 P50/P95 延迟; - 并发吞吐; - 峰值内存/显存; - 空白图和异常图片行为。 推荐顺序: 1. 固定 RapidOCR、ONNXRuntime 和模型文件; 2. 确认实际 Execution Provider; 3. 调输入尺寸(`Global.max_side_len`、`Det.limit_side_len`); 4. 调检测(`box_thresh`、`thresh`、`unclip_ratio`); 5. 调最终过滤(`text_score`); 6. 调批量与线程以优化性能; 7. 最后比较不同模型和后端。 每次只改一到两个相关参数,并保存对照结果。 --- ## 20. 本项目的优先建议 1. 调参只修改项目根 `.env` 中已开放的 `OCR_*` 变量,并重启服务。 2. 输入方向稳定时可关闭 `OCR_USE_CLS`,对比准确率和延迟后决定。 3. API 当前消费细粒度坐标,因此保留 `return_word_box=True`;若未来只需要文本行框,可再单独评估关闭。 4. 不要向并发请求开放运行时阈值修改:3.8.1 会持久修改共享引擎状态。 5. DirectML 以实际 Provider 为准,不要只依据应用日志中的设备名称。 6. 升级 RapidOCR 前做模型回归;3.9.x 的默认模型、枚举和参数契约已有变化。 --- ## 21. 参考来源 官方资料: - [RapidOCR 参数说明](https://rapidai.github.io/RapidOCRDocs/main/install_usage/rapidocr/parameters/) - [RapidOCR 使用说明](https://rapidai.github.io/RapidOCRDocs/main/install_usage/rapidocr/usage/) - [RapidOCR 推理引擎说明](https://rapidai.github.io/RapidOCRDocs/main/install_usage/rapidocr/how_to_use_infer_engine/) - [RapidOCR 模型列表](https://rapidai.github.io/RapidOCRDocs/latest/model_list/) - [RapidOCR GitHub 仓库](https://github.com/RapidAI/RapidOCR) - [RapidOCR Releases](https://github.com/RapidAI/RapidOCR/releases) - [ONNXRuntime DirectML Execution Provider](https://onnxruntime.ai/docs/execution-providers/DirectML-ExecutionProvider.html) - [ONNXRuntime CUDA Execution Provider](https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html) 本地核验文件: - `.venv/Lib/site-packages/rapidocr/config.yaml` - `.venv/Lib/site-packages/rapidocr/main.py` - `.venv/Lib/site-packages/rapidocr/utils/parse_parameters.py` - `.venv/Lib/site-packages/rapidocr/utils/typings.py` - `.venv/Lib/site-packages/rapidocr/inference_engine/onnxruntime/main.py` - `.venv/Lib/site-packages/rapidocr/inference_engine/onnxruntime/provider_config.py` - `config/config.py` - `engine/ocr_engine.py` - `.env` 对于本文未给出明确结论的第三方模型兼容性、具体后端性能和模型元数据,应以实际模型来源、对应版本源码和业务样本验证为准。