rapid-structure-rs 1.0.0

Rust document-structure engine for PaddleOCR-family ONNX models: text detection/recognition (PP-OCRv6), table structure (SLANet-plus) and formula recognition (PP-FormulaNet) on Windows x64 via ONNX Runtime.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

use std::path::PathBuf;

use serde::{Deserialize, Serialize};

use crate::config::{LangRec, ModelType, OcrVersion};

/// 普通 OCR 识别模型配置：模型文件、字典、版本与下载策略。
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(default, deny_unknown_fields)]
pub struct ModelConfig {
    pub lang: LangRec,
    pub ocr_version: OcrVersion,
    pub model_type: ModelType,
    pub model_path: Option<PathBuf>,
    pub rec_keys_path: Option<PathBuf>,
    pub allow_download: bool,
}

impl Default for ModelConfig {
    fn default() -> Self {
        Self {
            lang: LangRec::default(),
            ocr_version: OcrVersion::default(),
            model_type: ModelType::default(),
            model_path: None,
            rec_keys_path: None,
            allow_download: true,
        }
    }
}

/// 普通 OCR 识别器配置（rec 阶段）。
///
/// 这里没有 `runtime` 字段：线程/provider/arena 由 `EngineConfig::runtime` 唯一表达，
/// 见 `runtime::profile`。
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(default, deny_unknown_fields)]
pub struct RecognizerConfig {
    pub model: ModelConfig,
    pub rec_batch_num: usize,
    pub rec_img_shape: [usize; 3],
    pub model_store_dir: Option<PathBuf>,
}

impl Default for RecognizerConfig {
    fn default() -> Self {
        Self {
            model: ModelConfig::default(),
            rec_batch_num: 6,
            rec_img_shape: [3, 48, 320],
            model_store_dir: None,
        }
    }
}

/// 普通 OCR 词级/单字框输出开关。
#[derive(Debug, Clone, Copy, Default, Serialize, Deserialize)]
pub struct RecognizeOptions {
    pub return_word_box: bool,
    pub return_single_char_box: bool,
}