Commit Graph
532 Commits
Author SHA1 Message Date
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
Mert dd6f0e4d0c fix: lazy load font (#586)
* make visres lazy

* move none check
2025-11-07 20:48:43 +08:00
SWHL 787819457f fix: fixed issue #566 (#573)
* chore: update files

* chore: update files

* chore: refacter torch config

* chore: refacter torch infersession

* chore: refacter torch infersession

* test: change image url from github to modelscope
2025-11-07 20:00:33 +08:00
SWHL 13fb8cc8db fix: fixed issue #542 (#564)
* fix: fixed issue #542

* test: update the image of the test cls

* test: update the ground truth of the test cls image
2025-10-11 21:40:00 +08:00
SWHL 32dad48ebb fix: fixed issue #555
* fix: fixed issue #555

* chore: refactor code

* chore: refactor code

* chore: refactor code

* chore: refactor code

* chore: refactor code
2025-09-30 17:44:20 +08:00
SWHL ff2050266e chore: refacter torch infersession 2025-09-28 21:19:48 +08:00
yuege969andyuege969 eb6c685616 The torch engine adds support for Ascend NPU (#557)
Co-authored-by: yuege969 <mingyue.zhou@senscape.com.cn>
2025-09-28 20:35:15 +08:00
SWHL dec60563b4 fix: fixed issue #554 2025-09-18 20:36:44 +08:00
SWHL 017b8fd5c7 feat: add en/el/th ppocr_v5_rec_mobile paddle/onnx/openvino (#546)
* feat: add en/el/th ppocr_v5_rec_mobile paddle/onnx/openvino

* test: change to a more secure code
2025-09-06 17:50:05 +08:00
SWHL a40454c9b0 fix: fixed issue #544 (#545) 2025-09-04 21:55:34 +08:00
SWHL 386c415933 fix: remove deprecated getLevalName 2025-09-04 08:29:15 +08:00
SWHL 336e6424a5 fix: fixed issue #531 2025-09-03 21:58:15 +08:00
SWHL a0bb926500 fix: fixed issue #528 2025-09-01 09:36:46 +08:00
SWHL af65796b78 fix: fixed issue #527 2025-09-01 08:28:10 +08:00
SWHL 6d1b451812 fix: fixed repeating printing in font logs 2025-08-01 09:55:45 +08:00