Commit Graph
546 Commits
Author SHA1 Message Date
SWHL d312f67f44 refactor: optimize to package whl logic 2026-06-09 23:04:42 +08:00
SWHL a4c7382bba feat: add default download model configuration 2026-06-09 23:04:08 +08:00
SWHL e8938d8c2a refactor: optimize import 2026-06-09 23:02:06 +08:00
SWHL 7b2d368ac6 fix: fixed dml typo 2026-05-22 14:45:35 +08:00
StarLineandCopilot Autofix powered by AI 00f71b2d5e feat: update to_markdown rule (#672)
* Update to_markdown.py

修改重排规则:
中文段落
表格/多栏
标题与正文混排
OCR 框轻微倾斜

* Potential fix for pull request finding

refactor: format function call for readability
- Wrap `cls.get_gap_text` call into multi-line format
- No functional changes, only code style improvement

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

---------

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
2026-05-18 14:39:11 +08:00
SWHL 3857ff5153 build: update omegaconf version constraint 2026-05-06 09:59:20 +08:00
SWHL 49b4fcb44c fix(cli): handle language option and empty visualization results
- pass --lang_type into RapidOCR recognition config
- skip CLI visualization when OCR or word results are empty
- validate parameter update keys before applying config updates
- add regression tests for CLI language, empty visualization, and invalid params
2026-04-27 16:51:56 +08:00
SWHL d48c4cf312 refactor: optimize model_root_dir logic 2026-04-24 11:24:49 +08:00
FeiyueHa 57dfac08d8 fix: check model_root_dir only if model_path is None (#665) 2026-04-24 10:18:43 +08:00
SWHL d4bf8f2b96 fix: add missing DEFAULT_MODEL_PATH property (#664) 2026-04-11 10:29:01 +08:00
SWHL abaaeb3043 chore: update default_model version to v3.8.0 2026-04-08 21:26:32 +08:00
SWHL 143f46242d fix: update paddle/cls/v5 model url 2026-04-08 21:17:07 +08:00
SWHL c5d1ea7f4a chore: update model name 2026-04-08 10:05:46 +08:00
SWHL ece967983d test: add cls unit testings 2026-04-07 18:47:47 +08:00
SWHL b73ec51ecf feat: add PP-OCRv5 cls module 2026-04-07 17:35:01 +08:00
SWHL 9023589c86 feat: add PP-OCRv5 cls module 2026-04-07 17:34:15 +08:00
SWHL 5ef452516d fix: added missing package import 2026-03-19 17:36:24 +08:00
SWHL 8ea96265b5 feat(python): add model_root_dir and download_models (API + CLI) 2026-03-19 17:27:44 +08:00
SWHL f7961d3d2c chore: update default_model url path 2026-03-04 23:17:44 +08:00
HighDoping 597fe059ac feat: add experimental MPS support to PyTorch backend. (#639) 2026-03-04 16:32:50 +08:00
SWHL 5aa9079e75 refactor: standardize public interface with __all__
Close issue #646
2026-02-28 23:37:25 +08:00
HighDoping 7c880e8ac9 feat(inference_engine): add experimental CoreML support for onnxruntime (#634)
* Add CoreML support for onnxruntime.

* Add more CoreML execution provider configs.
2026-02-28 18:25:00 +08:00
SWHL 9d12451b7b refactor(openvino): remove dict_url from model configs 2026-02-26 19:42:45 +08:00
SWHL 80bbdc9615 test(tensorrt): skip TensorRT tests when not installed and exclude from parametrized suites 2026-02-26 17:34:30 +08:00
SWHL e549f7eacc feat(inference): add get_character_list abstract method to InferSession 2026-02-26 17:32:47 +08:00
SWHL 106d4d189d fix(openvino): fixed the missing List import 2026-02-26 17:07:35 +08:00
SWHL 9108b617ab feat(openvino): support retrieving character list from model runtime info 2026-02-25 19:00:08 +08:00
locviettriandCursor b132a915e7 feat(inference_engine): add TensorRT engine support (#623)
* feat(inference_engine): add TensorRT engine support

- Add TensorRT inference session with dynamic shape support
- Implement automatic engine building and caching per GPU architecture
- Add pre-allocated buffers with pinned memory optimization
- Support FP16/FP32/INT8 precision modes
- Add optimization profiles for det/rec/cls models
- Include comprehensive error handling and resource cleanup
- Add unit tests for TensorRT engine (skipped when unavailable)
- Add TensorRT configuration section in config.yaml

* fix(tensorrt): dynamically allocate output buffers to prevent overflow

* Update test_engine.py with TensorRT support final version

* feat(tensorrt): add model_type to engine cache key

- Include model_type in TensorRT engine cache key generation
- Ensures proper cache differentiation based on model_type

* fix(tensorrt): update maximum shape configurations for input and output tensors

- Increased max_shape values in config.yaml and engine_builder.py to 2048 for both detection and recognition tasks.
- Updated default maximum shapes in memory_utils.py to reflect the new limits for input and output buffers.

* feat(tensorrt): implement square input handling for MULTI models

- Added functionality to check if the model requires square input based on configuration.
- Implemented padding of input to square shape and cropping of output back to original dimensions.
- Introduced methods to determine maximum profile size and handle square input requirements for TensorRT inference.

---------

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-02-05 08:09:21 +08:00
SWHL 009a210a79 chore: update default image url 2026-02-01 14:36:32 +08:00
SWHL 75d0d159d2 chore: update model list to direct version 2026-01-28 22:32:35 +08:00
SWHL 3a4a91020b test(mnn): add mnn unit testings 2026-01-28 22:18:58 +08:00
SWHL 0ae0375710 feat(mnn): add mnn model list 2026-01-28 22:18:05 +08:00
SWHL 58bf4f81cc refactor: add tip when model_dict is None
Closes #628
2026-01-28 22:11:01 +08:00
SWHL 402aadb121 refactor(openvino): fix openvion error typo error 2026-01-28 21:40:08 +08:00
locviettri 4818b4f1e1 refactor: modularize MNN and OpenVINO Inference Engines (#629)
* refactor: modularize mnn and openvino inference engines

* Revert test_engine.py to original version
2026-01-28 21:35:15 +08:00
narcissus 86c527e14b feat(engine): 引入 MNN 推理后端,丰富部署生态 (#613) 2026-01-22 09:41:41 +08:00
moesnow 4c39ce92fd fix: replace deprecated openvino.runtime import (#626) 2026-01-21 08:17:41 +08:00
SWHL 7d88c84acf fix(inference_engine): fix model type comparison and improve error handling
Closed #565
2026-01-06 21:57:34 +08:00
SWHL 5cd7de1e47 refactor: refactor paddle inference code (#622)
* refactor: refactor paddle inference code

* fix(paddle): correct typo and improve device env setup
2026-01-05 21:19:25 +08:00
SWHL 09dbaf5ae5 refactor: remove unused parameters from calc_word_boxes 2025-12-26 22:07:10 +08:00
SWHL 32c705b99c fix: fixed error of showing cls result (#621)
* fix: fixed error of showing cls result

Related #616

* refactor(cls/utils): use cv2.ROTATE_180 and add type hint
2025-12-26 21:51:16 +08:00
SWHL 8487133fc5 feat: update five multi language files (#619)
* feat: update five multi language files

* feat(rec): enable RTL text display for Arabic

* test(rec): add arabic, cyrillic, devanagari, ta and te unit testings

* test(rec): remove duplicate cyrillic unit testing

* feat(rec): update url to v3.5.0
2025-12-26 21:24:58 +08:00
SWHL df8c571f1c fix: correct incorrect character coordinates in returned data (#614)
* chore: update files

* fix: correct incorrect character coordinates in returned data

Related #612

* chore: restore demo

* fix(utils): add missing type annotations to filter_by_indices and is_chinese_char
2025-12-18 08:20:43 +08:00
SWHL 6ff07a57f8 fix: sync filtering of scores with word_results
Closes #608
2025-12-09 21:29:26 +08:00
SWHL c65d816b5b feat: replace flawed RGBA-to-BGA conversion with auto-contrast background blending 2025-12-09 21:11:44 +08:00
SWHL 4e0224c395 fix: fixed issue #589 2025-12-02 21:42:48 +08:00
SWHL 7c1e4601b8 fix: fixed boxes sorting error 2025-12-02 11:26:16 +08:00
Mert 3f3fb13235 feat: optimized box sorting (#587)
* optimized box sorting

* pr feedback
2025-12-02 09:52:33 +08:00
SWHL d2e9e6b844 refacter: optimize the parameter update method 2025-12-01 17:57:40 +08:00
WangFeng Huang 3be5925821 feat(python/rapidocr): Default parameters respect configuration file settings. (#591)
- Implement conditional assignments to only update instance attributes when parameters are explicitly provided
2025-12-01 17:04:03 +08:00