Commit Graph
523 Commits
Author SHA1 Message Date
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
SWHL 480abaf854 chore: remove rapidocr_paddle/torch/onnxruntime/openvion code 2025-07-29 08:08:48 +08:00
SWHL 66e00f4beb feat(rapidocr): update ppocrv5 minor language config and unit testings 2025-07-27 18:02:19 +08:00
SWHL c50038bc1e feat(rapidocr): add korean ppocrv5 config and update uniting testing 2025-07-27 17:54:35 +08:00
SWHL 78b39ca378 feat(rapidocr): support ppocr v5 minor language model(korean / latin / eslav) (#516)
* feat(rapidocr): add korean ppocrv5 config and update uniting testing

* feat: support latin ppocrv5 model(paddle/ort)

* feat: support eslav ppocrv5 model(paddle/ort)

* feat(rapidocr): update model version
2025-07-27 17:51:02 +08:00
SWHL 4c4bcab0be feat(rapidocr): add font_path and lang_type parameter of cli 2025-07-27 09:15:12 +08:00
SWHL 20cbc41e36 test: add to_json and to_markdown unit testings 2025-07-27 08:52:35 +08:00
SWHL 1a94bb1935 feat: support to_json func (fixed issue #489) 2025-07-24 23:02:36 +08:00
SWHL c46d0125f3 chore: update files (#513) 2025-07-24 21:13:24 +08:00
SWHL 1797850c15 feat(rapidocr): add ppocr v5 torch version (#511)
* feat: support torch ppocrv5 det

* feat: support torch ppocrv5 rec

* chore: update files

* chore: update files

* test: add ppocrv5 torch unit testings

* fix: fix typos
2025-07-23 22:08:31 +08:00