Commit Graph
100 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
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 3a4a91020b test(mnn): add mnn unit testings 2026-01-28 22:18:58 +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 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 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 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 336e6424a5 fix: fixed issue #531 2025-09-03 21:58:15 +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 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
SWHL 0b2d7f0d24 test: update unit testing when adjusting the format of returning word box 2025-06-18 08:22:56 +08:00
SWHL 0190646256 fix(rapidoicr): fixed issue #446 (#453)
* fix(rapidoicr): fixed issue #446

* test(rapidocr): remove pdb code
2025-06-10 21:45:15 +08:00
SWHL 41d5824e13 feat: adapt PR #451
* feat: adapt CANN of onnxruntime (PR #428)

* chore: fix typos

* docs: update README

* fix(rapidocr): fixed issue #438

* chore: update files

* chore: update files

* chore: update files
2025-06-10 20:36:41 +08:00
SWHL dcbe0d7efa test: add unit testings of issue 440 (#445) 2025-06-06 22:42:41 +08:00
SWHL 5f86f514dd feat: support PP-OCRv5 (#437)
* chore: add v5 det model path

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files
2025-06-01 10:50:02 +08:00
SWHL 4d35ed272a tests(rapidocr): update tests of ch_doc_server 2025-05-16 19:39:47 +08:00
SWHL 80c72f54e9 feat(rapidocr): integrated PP-OCRv4_dervr_rec-d oc model (#426)
* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files
2025-05-16 09:27:01 +08:00
SWHL 5776a55e8f feat(return_word_box): support word box of English and number text (#423)
* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: update files

* chore: format code

* chore: format code

* chore: format code
2025-05-13 20:28:40 +08:00
SWHL fbaaaba486 fix(rapidocr): fixed issue #394 (#411)
* chore(rapidocr): add model SHA256

* chore(rapidocr): add model SHA256

* docs: update online demo link

* chore: update model path

* chore: update files

* chore: update files

* chore: update files

* fix(rapidocr): fixed paramter of DownloadFile
2025-04-25 20:39:38 +08:00
SWHLandCopilot d48f31e815 fix(rapidocr): fixed issue #381 #377 (#384)
* chore: update files

* fix(rapidocr): fixed issue #381

* chore: update demo

* Update python/setup.py

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

---------

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
2025-04-03 22:01:34 +08:00
SWHL b256f83316 fix(vis_result): support to download direct font file 2025-03-31 21:11:14 +08:00
SWHL 992cdf9b8b test(rapidocr): update vis unit tests 2025-03-26 09:58:16 +08:00
SWHL 202dfb91fc feat(rapidocr): support to other language ocr and update visualization (#369)
* fix(vis_result): support to download direct font file

* chore: update files

* feat(rapidocr): update vis

* feat(rapidocr): update vis
2025-03-26 09:41:23 +08:00
SWHL 47418c2005 feat(rapidocr): support img url as input and optimize vis (#356)
* feat(rapidocr): support image url input

* feat(rapidocr): support img url as input and optimize vis
2025-03-12 09:18:08 +08:00
SWHL 2a420e013e fix(rapidocr): complete parsing parameters 2025-03-09 10:06:04 +08:00
SWHL 4beb30ec79 feat: optimize server model 2025-03-04 09:44:16 +08:00
SWHL 246cb1c321 chore: update files 2025-02-18 22:35:32 +08:00
SWHL f7c0ab1fc0 chore: update files 2025-02-18 09:42:44 +08:00
SWHL e044c3fa5e chore: update files 2025-02-17 22:25:21 +08:00
SWHL 6de7194647 chore: update files 2025-02-14 14:13:48 +08:00
SWHL 6a2b9e5034 chore: update files 2025-02-13 22:22:01 +08:00
SWHL 6189f611b6 chore: update files 2025-02-13 10:37:25 +08:00
SWHL afbb6cbfd3 chore: update files 2025-02-12 10:26:43 +08:00
SWHL 5bf7702915 chore: update files 2025-02-11 23:15:44 +08:00
SWHL f10eab5a36 chore: update files 2025-02-11 14:30:01 +08:00
Joker1212 9a8a4a3916 suport torch inference (#320)
* feat: add rapidocr_torch inference

* ci: fix push workflow
2025-02-08 09:52:34 +08:00
SWHL 12ea82affe chore: update files 2025-02-06 09:50:29 +08:00
SWHL 1518e72235 chore: update files 2025-01-20 21:18:56 +08:00
SWHL 94576e5e4c chore: optimize code 2025-01-17 09:36:38 +08:00
SWHL c7c8bfb305 feat(rapidocr_paddle): support en char rec 2024-12-04 08:12:16 +08:00
SWHL ea003db3d6 feat(rapidocr_openvino): support en char rec 2024-12-03 22:22:27 +08:00
SWHL cababcf25c feat(rapidocr_openvino): support en char rec 2024-12-03 22:15:25 +08:00
Joker1212 01a5a65952 feat(rapidocr_onnxruntime): support en char rec (#272)
* feat: sup en char rec

* test: add en char rec test
2024-12-03 09:30:54 +08:00