Skip to main content

oar_ocr_core/utils/
mod.rs

1//! Utility functions for the OCR pipeline.
2//!
3//! This module provides various utility functions used throughout the OCR pipeline,
4//! including image processing helpers, tensor conversion functions, cropping helpers,
5//! classification utilities, and logging setup.
6
7pub mod bbox_crop;
8pub mod cow;
9pub mod crop;
10pub mod dict;
11pub mod image;
12pub mod tensor;
13pub mod topk;
14pub mod transform;
15pub mod validation;
16
17// Re-export image processing functions
18pub use image::{
19    OCRResizePadConfig, PaddingStrategy, ResizePadConfig, calculate_center_crop_coords,
20    check_image_size, create_rgb_image, dynamic_to_gray, dynamic_to_rgb, load_image, load_images,
21    load_images_batch_with_policy, load_images_batch_with_threshold, mask_region, mask_regions,
22    ocr_resize_and_pad, pad_image, resize_and_pad, resize_gray_image, resize_image,
23    resize_images_batch, resize_images_batch_to_dynamic, rgb_to_grayscale, slice_gray_image,
24    slice_image, validate_crop_bounds,
25};
26
27// Re-export tensor utility functions
28pub use tensor::*;
29
30// Re-export transform utility functions
31pub use transform::get_rotate_crop_image;
32
33// Re-export shared processors-style utilities
34pub use bbox_crop::BBoxCrop;
35pub use crop::Crop;
36pub use topk::{Topk, TopkResult};
37
38// Re-export validation utilities
39pub use validation::{
40    ScoreValidator, validate_length_match, validate_max_value, validate_positive_dimensions,
41};
42
43// Re-export dictionary utilities
44pub use dict::{read_character_dict, read_dict_content, require_path};